From efc389018ef321e913ebeaf5214a3f64e29d1067 Mon Sep 17 00:00:00 2001 From: August <90501900+HUANGYming@users.noreply.github.com> Date: Mon, 20 Nov 2023 11:42:31 +0800 Subject: [PATCH] Update environment from CUDA10.2 to CUDA 11.8 Hello author, I have been using the A100 to train the ISNet model. However, I encountered a problem with CUDA incompatibility. To resolve this issue, I spent some time upgrading the CUDA-related packages and conducting compatibility checks. Now, ISNet can run on newer architecture GPUs like the A100/4090. I have exported the upgraded environment configuration to share with you, hoping to help more people avoid the pain of environment upgrading. The two environment configuration files I am providing are compatible with CUDA 11.8. Using this environment, ISNet can run on GPUs with Ampere architecture and earlier, such as the 30 series cards, 40 series cards, A100, A10, etc. Except for the H100, which requires a CUDA 12+ environment, CUDA 11.8 currently supports the vast majority of Nvidia graphics cards. --- IS-Net/environment_cu118.yaml | 188 ++++++++++++++++++++++++++++++++++ IS-Net/requirements_cu118.txt | 183 +++++++++++++++++++++++++++++++++ 2 files changed, 371 insertions(+) create mode 100644 IS-Net/environment_cu118.yaml create mode 100644 IS-Net/requirements_cu118.txt diff --git a/IS-Net/environment_cu118.yaml b/IS-Net/environment_cu118.yaml new file mode 100644 index 0000000..e7826c7 --- /dev/null +++ b/IS-Net/environment_cu118.yaml @@ -0,0 +1,188 @@ +name: ISNet-cu118 +channels: + - pytorch + - nvidia + - anaconda + - defaults +dependencies: + - _libgcc_mutex=0.1=main + - _openmp_mutex=5.1=1_gnu + - aom=3.6.0=h6a678d5_0 + - blas=1.0=mkl + - blosc=1.21.3=h6a678d5_0 + - brotli=1.0.9=h5eee18b_7 + - brotli-bin=1.0.9=h5eee18b_7 + - brotli-python=1.0.9=py38h6a678d5_7 + - brunsli=0.1=h2531618_0 + - bzip2=1.0.8=h7b6447c_0 + - c-ares=1.19.1=h5eee18b_0 + - ca-certificates=2023.08.22=h06a4308_0 + - certifi=2023.7.22=py38h06a4308_0 + - cffi=1.15.0=py38h7f8727e_0 + - cfitsio=3.470=h5893167_7 + - charls=2.2.0=h2531618_0 + - charset-normalizer=2.0.4=pyhd3eb1b0_0 + - click=8.1.7=py38h06a4308_0 + - cloudpickle=2.2.1=py38h06a4308_0 + - contourpy=1.0.5=py38hdb19cb5_0 + - cryptography=41.0.3=py38h130f0dd_0 + - cuda-cudart=11.8.89=0 + - cuda-cupti=11.8.87=0 + - cuda-libraries=11.8.0=0 + - cuda-nvrtc=11.8.89=0 + - cuda-nvtx=11.8.86=0 + - cuda-runtime=11.8.0=0 + - cudatoolkit=11.8.0=h6a678d5_0 + - cycler=0.11.0=pyhd3eb1b0_0 + - cytoolz=0.12.0=py38h5eee18b_0 + - dask-core=2023.4.1=py38h06a4308_0 + - dav1d=1.2.1=h5eee18b_0 + - dbus=1.13.18=hb2f20db_0 + - expat=2.5.0=h6a678d5_0 + - ffmpeg=4.3=hf484d3e_0 + - fftw=3.3.9=h27cfd23_1 + - filelock=3.9.0=py38h06a4308_0 + - fontconfig=2.14.1=h52c9d5c_1 + - fonttools=4.25.0=pyhd3eb1b0_0 + - freetype=2.12.1=h4a9f257_0 + - fsspec=2023.9.2=py38h06a4308_0 + - giflib=5.2.1=h5eee18b_3 + - glib=2.63.1=h5a9c865_0 + - gmp=6.2.1=h295c915_3 + - gmpy2=2.1.2=py38heeb90bb_0 + - gnutls=3.6.15=he1e5248_0 + - gst-plugins-base=1.14.0=hbbd80ab_1 + - gstreamer=1.14.0=hb453b48_1 + - icu=58.2=he6710b0_3 + - idna=3.4=py38h06a4308_0 + - imagecodecs=2023.1.23=py38hc4b7b5f_0 + - imageio=2.31.4=py38h06a4308_0 + - importlib-metadata=6.0.0=py38h06a4308_0 + - importlib_resources=6.1.0=py38h06a4308_0 + - intel-openmp=2021.4.0=h06a4308_3561 + - jinja2=3.1.2=py38h06a4308_0 + - jpeg=9e=h5eee18b_1 + - jxrlib=1.1=h7b6447c_2 + - kiwisolver=1.4.4=py38h6a678d5_0 + - krb5=1.20.1=h568e23c_1 + - lame=3.100=h7b6447c_0 + - lazy_loader=0.3=py38h06a4308_0 + - lcms2=2.12=h3be6417_0 + - lerc=3.0=h295c915_0 + - libaec=1.0.4=he6710b0_1 + - libavif=0.11.1=h5eee18b_0 + - libbrotlicommon=1.0.9=h5eee18b_7 + - libbrotlidec=1.0.9=h5eee18b_7 + - libbrotlienc=1.0.9=h5eee18b_7 + - libcublas=11.11.3.6=0 + - libcufft=10.9.0.58=0 + - libcufile=1.8.1.2=0 + - libcurand=10.3.4.101=0 + - libcurl=7.88.1=h91b91d3_2 + - libcusolver=11.4.1.48=0 + - libcusparse=11.7.5.86=0 + - libdeflate=1.17=h5eee18b_1 + - libedit=3.1.20221030=h5eee18b_0 + - libev=4.33=h7f8727e_1 + - libffi=3.2.1=hf484d3e_1007 + - libgcc-ng=11.2.0=h1234567_1 + - libgfortran-ng=11.2.0=h00389a5_1 + - libgfortran5=11.2.0=h1234567_1 + - libgomp=11.2.0=h1234567_1 + - libiconv=1.16=h7f8727e_2 + - libidn2=2.3.4=h5eee18b_0 + - libjpeg-turbo=2.0.0=h9bf148f_0 + - libnghttp2=1.52.0=ha637b67_1 + - libnpp=11.8.0.86=0 + - libnvjpeg=11.9.0.86=0 + - libpng=1.6.39=h5eee18b_0 + - libssh2=1.10.0=h37d81fd_2 + - libstdcxx-ng=11.2.0=h1234567_1 + - libtasn1=4.19.0=h5eee18b_0 + - libtiff=4.5.1=h6a678d5_0 + - libunistring=0.9.10=h27cfd23_0 + - libuuid=1.41.5=h5eee18b_0 + - libwebp=1.3.2=h11a3e52_0 + - libwebp-base=1.3.2=h5eee18b_0 + - libxcb=1.15=h7f8727e_0 + - libxml2=2.9.14=h74e7548_0 + - libzopfli=1.0.3=he6710b0_0 + - llvm-openmp=14.0.6=h9e868ea_0 + - locket=1.0.0=py38h06a4308_0 + - lz4-c=1.9.4=h6a678d5_0 + - markupsafe=2.1.1=py38h7f8727e_0 + - matplotlib=3.7.2=py38h06a4308_0 + - matplotlib-base=3.7.2=py38h1128e8f_0 + - mkl=2021.4.0=h06a4308_640 + - mkl-service=2.4.0=py38h7f8727e_0 + - mkl_fft=1.3.1=py38hd3c417c_0 + - mkl_random=1.2.2=py38h51133e4_0 + - mpc=1.1.0=h10f8cd9_1 + - mpfr=4.0.2=hb69a4c5_1 + - mpmath=1.3.0=py38h06a4308_0 + - munkres=1.1.4=py_0 + - ncurses=6.4=h6a678d5_0 + - nettle=3.7.3=hbbd107a_1 + - networkx=3.1=py38h06a4308_0 + - numpy=1.24.3=py38h14f4228_0 + - numpy-base=1.24.3=py38h31eccc5_0 + - openh264=2.1.1=h4ff587b_0 + - openjpeg=2.4.0=h3ad879b_0 + - openssl=1.1.1w=h7f8727e_0 + - packaging=23.1=py38h06a4308_0 + - partd=1.4.1=py38h06a4308_0 + - pcre=8.45=h295c915_0 + - pillow=10.0.1=py38ha6cbd5a_0 + - pip=23.3=py38h06a4308_0 + - pycparser=2.21=pyhd3eb1b0_0 + - pyopenssl=23.2.0=py38h06a4308_0 + - pyparsing=3.0.9=py38h06a4308_0 + - pyqt=5.9.2=py38h05f1152_4 + - pysocks=1.7.1=py38h06a4308_0 + - python=3.8.0=h0371630_2 + - python-dateutil=2.8.2=pyhd3eb1b0_0 + - pytorch=2.1.1=py3.8_cuda11.8_cudnn8.7.0_0 + - pytorch-cuda=11.8=h7e8668a_5 + - pytorch-mutex=1.0=cuda + - pywavelets=1.4.1=py38h5eee18b_0 + - pyyaml=6.0.1=py38h5eee18b_0 + - qt=5.9.7=h5867ecd_1 + - readline=7.0=h7b6447c_5 + - requests=2.31.0=py38h06a4308_0 + - scikit-image=0.20.0=py38h6a678d5_0 + - scipy=1.9.1=py38h14f4228_0 + - setuptools=68.0.0=py38h06a4308_0 + - sip=4.19.13=py38h295c915_0 + - six=1.16.0=pyhd3eb1b0_1 + - snappy=1.1.9=h295c915_0 + - sqlite=3.33.0=h62c20be_0 + - sympy=1.11.1=py38h06a4308_0 + - tifffile=2023.4.12=py38h06a4308_0 + - tk=8.6.12=h1ccaba5_0 + - toolz=0.12.0=py38h06a4308_0 + - torchaudio=2.1.1=py38_cu118 + - torchtriton=2.1.0=py38 + - torchvision=0.16.1=py38_cu118 + - tornado=6.3.3=py38h5eee18b_0 + - tqdm=4.65.0=py38hb070fc8_0 + - typing_extensions=4.7.1=py38h06a4308_0 + - urllib3=1.26.18=py38h06a4308_0 + - wheel=0.41.2=py38h06a4308_0 + - xz=5.4.2=h5eee18b_0 + - yaml=0.2.5=h7b6447c_0 + - zfp=1.0.0=h6a678d5_0 + - zipp=3.11.0=py38h06a4308_0 + - zlib=1.2.13=h5eee18b_0 + - zstd=1.5.5=hc292b87_0 + - pip: + - appdirs==1.4.4 + - conda-pack==0.7.1 + - docker-pycreds==0.4.0 + - gitdb==4.0.11 + - gitpython==3.1.40 + - protobuf==4.25.1 + - psutil==5.9.6 + - sentry-sdk==1.35.0 + - setproctitle==1.3.3 + - smmap==5.0.1 + - wandb==0.16.0 diff --git a/IS-Net/requirements_cu118.txt b/IS-Net/requirements_cu118.txt new file mode 100644 index 0000000..a15c4ad --- /dev/null +++ b/IS-Net/requirements_cu118.txt @@ -0,0 +1,183 @@ +# This file may be used to create an environment using: +# $ conda create --name --file +# platform: linux-64 +_libgcc_mutex=0.1=main +_openmp_mutex=5.1=1_gnu +aom=3.6.0=h6a678d5_0 +appdirs=1.4.4=pypi_0 +blas=1.0=mkl +blosc=1.21.3=h6a678d5_0 +brotli=1.0.9=h5eee18b_7 +brotli-bin=1.0.9=h5eee18b_7 +brotli-python=1.0.9=py38h6a678d5_7 +brunsli=0.1=h2531618_0 +bzip2=1.0.8=h7b6447c_0 +c-ares=1.19.1=h5eee18b_0 +ca-certificates=2023.08.22=h06a4308_0 +certifi=2023.7.22=py38h06a4308_0 +cffi=1.15.0=py38h7f8727e_0 +cfitsio=3.470=h5893167_7 +charls=2.2.0=h2531618_0 +charset-normalizer=2.0.4=pyhd3eb1b0_0 +click=8.1.7=py38h06a4308_0 +cloudpickle=2.2.1=py38h06a4308_0 +conda-pack=0.7.1=pypi_0 +contourpy=1.0.5=py38hdb19cb5_0 +cryptography=41.0.3=py38h130f0dd_0 +cuda-cudart=11.8.89=0 +cuda-cupti=11.8.87=0 +cuda-libraries=11.8.0=0 +cuda-nvrtc=11.8.89=0 +cuda-nvtx=11.8.86=0 +cuda-runtime=11.8.0=0 +cudatoolkit=11.8.0=h6a678d5_0 +cycler=0.11.0=pyhd3eb1b0_0 +cytoolz=0.12.0=py38h5eee18b_0 +dask-core=2023.4.1=py38h06a4308_0 +dav1d=1.2.1=h5eee18b_0 +dbus=1.13.18=hb2f20db_0 +docker-pycreds=0.4.0=pypi_0 +expat=2.5.0=h6a678d5_0 +ffmpeg=4.3=hf484d3e_0 +fftw=3.3.9=h27cfd23_1 +filelock=3.9.0=py38h06a4308_0 +fontconfig=2.14.1=h52c9d5c_1 +fonttools=4.25.0=pyhd3eb1b0_0 +freetype=2.12.1=h4a9f257_0 +fsspec=2023.9.2=py38h06a4308_0 +giflib=5.2.1=h5eee18b_3 +gitdb=4.0.11=pypi_0 +gitpython=3.1.40=pypi_0 +glib=2.63.1=h5a9c865_0 +gmp=6.2.1=h295c915_3 +gmpy2=2.1.2=py38heeb90bb_0 +gnutls=3.6.15=he1e5248_0 +gst-plugins-base=1.14.0=hbbd80ab_1 +gstreamer=1.14.0=hb453b48_1 +icu=58.2=he6710b0_3 +idna=3.4=py38h06a4308_0 +imagecodecs=2023.1.23=py38hc4b7b5f_0 +imageio=2.31.4=py38h06a4308_0 +importlib-metadata=6.0.0=py38h06a4308_0 +importlib_resources=6.1.0=py38h06a4308_0 +intel-openmp=2021.4.0=h06a4308_3561 +jinja2=3.1.2=py38h06a4308_0 +jpeg=9e=h5eee18b_1 +jxrlib=1.1=h7b6447c_2 +kiwisolver=1.4.4=py38h6a678d5_0 +krb5=1.20.1=h568e23c_1 +lame=3.100=h7b6447c_0 +lazy_loader=0.3=py38h06a4308_0 +lcms2=2.12=h3be6417_0 +lerc=3.0=h295c915_0 +libaec=1.0.4=he6710b0_1 +libavif=0.11.1=h5eee18b_0 +libbrotlicommon=1.0.9=h5eee18b_7 +libbrotlidec=1.0.9=h5eee18b_7 +libbrotlienc=1.0.9=h5eee18b_7 +libcublas=11.11.3.6=0 +libcufft=10.9.0.58=0 +libcufile=1.8.1.2=0 +libcurand=10.3.4.101=0 +libcurl=7.88.1=h91b91d3_2 +libcusolver=11.4.1.48=0 +libcusparse=11.7.5.86=0 +libdeflate=1.17=h5eee18b_1 +libedit=3.1.20221030=h5eee18b_0 +libev=4.33=h7f8727e_1 +libffi=3.2.1=hf484d3e_1007 +libgcc-ng=11.2.0=h1234567_1 +libgfortran-ng=11.2.0=h00389a5_1 +libgfortran5=11.2.0=h1234567_1 +libgomp=11.2.0=h1234567_1 +libiconv=1.16=h7f8727e_2 +libidn2=2.3.4=h5eee18b_0 +libjpeg-turbo=2.0.0=h9bf148f_0 +libnghttp2=1.52.0=ha637b67_1 +libnpp=11.8.0.86=0 +libnvjpeg=11.9.0.86=0 +libpng=1.6.39=h5eee18b_0 +libssh2=1.10.0=h37d81fd_2 +libstdcxx-ng=11.2.0=h1234567_1 +libtasn1=4.19.0=h5eee18b_0 +libtiff=4.5.1=h6a678d5_0 +libunistring=0.9.10=h27cfd23_0 +libuuid=1.41.5=h5eee18b_0 +libwebp=1.3.2=h11a3e52_0 +libwebp-base=1.3.2=h5eee18b_0 +libxcb=1.15=h7f8727e_0 +libxml2=2.9.14=h74e7548_0 +libzopfli=1.0.3=he6710b0_0 +llvm-openmp=14.0.6=h9e868ea_0 +locket=1.0.0=py38h06a4308_0 +lz4-c=1.9.4=h6a678d5_0 +markupsafe=2.1.1=py38h7f8727e_0 +matplotlib=3.7.2=py38h06a4308_0 +matplotlib-base=3.7.2=py38h1128e8f_0 +mkl=2021.4.0=h06a4308_640 +mkl-service=2.4.0=py38h7f8727e_0 +mkl_fft=1.3.1=py38hd3c417c_0 +mkl_random=1.2.2=py38h51133e4_0 +mpc=1.1.0=h10f8cd9_1 +mpfr=4.0.2=hb69a4c5_1 +mpmath=1.3.0=py38h06a4308_0 +munkres=1.1.4=py_0 +ncurses=6.4=h6a678d5_0 +nettle=3.7.3=hbbd107a_1 +networkx=3.1=py38h06a4308_0 +numpy=1.24.3=py38h14f4228_0 +numpy-base=1.24.3=py38h31eccc5_0 +openh264=2.1.1=h4ff587b_0 +openjpeg=2.4.0=h3ad879b_0 +openssl=1.1.1w=h7f8727e_0 +packaging=23.1=py38h06a4308_0 +partd=1.4.1=py38h06a4308_0 +pcre=8.45=h295c915_0 +pillow=10.0.1=py38ha6cbd5a_0 +pip=23.3=py38h06a4308_0 +protobuf=4.25.1=pypi_0 +psutil=5.9.6=pypi_0 +pycparser=2.21=pyhd3eb1b0_0 +pyopenssl=23.2.0=py38h06a4308_0 +pyparsing=3.0.9=py38h06a4308_0 +pyqt=5.9.2=py38h05f1152_4 +pysocks=1.7.1=py38h06a4308_0 +python=3.8.0=h0371630_2 +python-dateutil=2.8.2=pyhd3eb1b0_0 +pytorch=2.1.1=py3.8_cuda11.8_cudnn8.7.0_0 +pytorch-cuda=11.8=h7e8668a_5 +pytorch-mutex=1.0=cuda +pywavelets=1.4.1=py38h5eee18b_0 +pyyaml=6.0.1=py38h5eee18b_0 +qt=5.9.7=h5867ecd_1 +readline=7.0=h7b6447c_5 +requests=2.31.0=py38h06a4308_0 +scikit-image=0.20.0=py38h6a678d5_0 +scipy=1.9.1=py38h14f4228_0 +sentry-sdk=1.35.0=pypi_0 +setproctitle=1.3.3=pypi_0 +setuptools=68.0.0=py38h06a4308_0 +sip=4.19.13=py38h295c915_0 +six=1.16.0=pyhd3eb1b0_1 +smmap=5.0.1=pypi_0 +snappy=1.1.9=h295c915_0 +sqlite=3.33.0=h62c20be_0 +sympy=1.11.1=py38h06a4308_0 +tifffile=2023.4.12=py38h06a4308_0 +tk=8.6.12=h1ccaba5_0 +toolz=0.12.0=py38h06a4308_0 +torchaudio=2.1.1=py38_cu118 +torchtriton=2.1.0=py38 +torchvision=0.16.1=py38_cu118 +tornado=6.3.3=py38h5eee18b_0 +tqdm=4.65.0=py38hb070fc8_0 +typing_extensions=4.7.1=py38h06a4308_0 +urllib3=1.26.18=py38h06a4308_0 +wandb=0.16.0=pypi_0 +wheel=0.41.2=py38h06a4308_0 +xz=5.4.2=h5eee18b_0 +yaml=0.2.5=h7b6447c_0 +zfp=1.0.0=h6a678d5_0 +zipp=3.11.0=py38h06a4308_0 +zlib=1.2.13=h5eee18b_0 +zstd=1.5.5=hc292b87_0