mirror of
https://github.com/xuebinqin/DIS.git
synced 2024-11-29 18:23:08 +01:00
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.
This commit is contained in:
August
GitHub
parent
ec4a4f4f8d
commit
efc389018e
188
IS-Net/environment_cu118.yaml
Normal file
188
IS-Net/environment_cu118.yaml
Normal file
@ -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
|
||||
183
IS-Net/requirements_cu118.txt
Normal file
183
IS-Net/requirements_cu118.txt
Normal file
@ -0,0 +1,183 @@
|
||||
# This file may be used to create an environment using:
|
||||
# $ conda create --name <env> --file <this 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
|
||||
Loading…
Reference in New Issue
Block a user