Hi,
I am trying to install pytorch preview version with cuda 11.8 on my conda virtual env,
but it is stuck at downloading pytorch-2.0.0.dev202 at 0%.
Is there anyone else facing this issue?
My environments are following:
OS : ubuntu 22.04
GPUs : 4090 x 4
NVIDIA-SMI : 525.85.05
CUDA :12
PYTHON : 3.8
ptrblck
February 23, 2023, 4:54am
2
Could you post the install command you were using so that I could double check the install routine?
1 Like
Oh, yeah sure. I used the following command, which is the one on the official website.
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch-nightly -c nvidia
I was able to download other packages (e.g. torchivision),
but the pytorch-2.0.0 keeps stuck.
ptrblck
February 23, 2023, 5:12am
4
Thanks for sharing the command! It looks correct and in case the download gets stuck you might be able to directly download the desired version from here and conda install
it from the local file.
I’ve just ran your provided command and while it also took some time to download the packages, it worked properly for me:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch-nightly -c nvidia
...
The following packages will be downloaded:
package | build
---------------------------|-----------------
cryptography-39.0.1 | py38h3d167d9_0 1.4 MB conda-forge
cuda-command-line-tools-11.8.0| 0 1 KB nvidia
cuda-cudart-11.8.89 | 0 197 KB nvidia
cuda-cupti-11.8.87 | 0 25.3 MB nvidia
cuda-libraries-11.8.0 | 0 1 KB nvidia
cuda-libraries-dev-11.3.0 | h3b286be_0 2 KB nvidia
cuda-nvrtc-11.8.89 | 0 19.1 MB nvidia
cuda-nvtx-11.8.86 | 0 57 KB nvidia
cuda-runtime-11.8.0 | 0 1 KB nvidia
cuda-thrust-11.4.43 | h00096a5_0 1.1 MB nvidia
cuda-tools-11.3.0 | h3b286be_0 2 KB nvidia
cuda-visual-tools-11.3.0 | h3b286be_0 2 KB nvidia
ffmpeg-5.1.2 | gpl_h8dda1f0_106 9.2 MB conda-forge
filelock-3.9.0 | pyhd8ed1ab_0 13 KB conda-forge
fontconfig-2.14.2 | h14ed4e7_0 266 KB conda-forge
jpeg-9e | h0b41bf4_3 235 KB conda-forge
libhwloc-2.9.0 | hd6dc26d_0 2.4 MB conda-forge
libva-2.17.0 | h0b41bf4_0 183 KB conda-forge
mpc-1.3.1 | hfe3b2da_0 114 KB conda-forge
openh264-2.3.1 | hcb278e6_2 702 KB conda-forge
pytorch-2.0.0.dev20230220 |py3.8_cuda11.8_cudnn8.7.0_0 1.41 GB pytorch-nightly
pytorch-cuda-11.8 | h7e8668a_3 7 KB pytorch-nightly
pytorch-mutex-1.0 | cuda 3 KB pytorch-nightly
svt-av1-1.4.1 | hcb278e6_0 2.4 MB conda-forge
tbb-2021.8.0 | hf52228f_0 1.4 MB conda-forge
torchaudio-2.0.0.dev20230223| py38_cu118 7.6 MB pytorch-nightly
torchtriton-2.0.0+c8bfe3f548| py38 62.6 MB pytorch-nightly
torchvision-0.15.0.dev20230222| py38_cu118 39.0 MB pytorch-nightly
urllib3-1.25.8 | py38h32f6830_1 160 KB conda-forge
xorg-libxext-1.3.4 | h0b41bf4_2 49 KB conda-forge
xorg-xextproto-7.3.0 | h0b41bf4_1003 30 KB conda-forge
------------------------------------------------------------
Total: 1.58 GB
...
Downloading and Extracting Packages
cryptography-39.0.1 | 1.4 MB | ############################################################## | 100%
cuda-libraries-dev-1 | 2 KB | ############################################################## | 100%
libhwloc-2.9.0 | 2.4 MB | ############################################################## | 100%
svt-av1-1.4.1 | 2.4 MB | ############################################################## | 100%
cuda-visual-tools-11 | 2 KB | ############################################################## | 100%
torchtriton-2.0.0+c8 | 62.6 MB | ############################################################## | 100%
filelock-3.9.0 | 13 KB | ############################################################## | 100%
torchaudio-2.0.0.dev | 7.6 MB | ############################################################## | 100%
cuda-command-line-to | 1 KB | ############################################################## | 100%
pytorch-2.0.0.dev202 | 1.41 GB | ############################################################## | 100%
fontconfig-2.14.2 | 266 KB | ############################################################## | 100%
xorg-libxext-1.3.4 | 49 KB | ############################################################## | 100%
urllib3-1.25.8 | 160 KB | ############################################################## | 100%
torchvision-0.15.0.d | 39.0 MB | ############################################################## | 100%
libva-2.17.0 | 183 KB | ############################################################## | 100%
cuda-nvrtc-11.8.89 | 19.1 MB | ############################################################## | 100%
xorg-xextproto-7.3.0 | 30 KB | ############################################################## | 100%
cuda-cupti-11.8.87 | 25.3 MB | ############################################################## | 100%
pytorch-mutex-1.0 | 3 KB | ############################################################## | 100%
cuda-cudart-11.8.89 | 197 KB | ############################################################## | 100%
cuda-nvtx-11.8.86 | 57 KB | ############################################################## | 100%
tbb-2021.8.0 | 1.4 MB | ############################################################## | 100%
pytorch-cuda-11.8 | 7 KB | ############################################################## | 100%
cuda-libraries-11.8. | 1 KB | ############################################################## | 100%
mpc-1.3.1 | 114 KB | ############################################################## | 100%
ffmpeg-5.1.2 | 9.2 MB | ############################################################## | 100%
cuda-tools-11.3.0 | 2 KB | ############################################################## | 100%
openh264-2.3.1 | 702 KB | ############################################################## | 100%
cuda-thrust-11.4.43 | 1.1 MB | ############################################################## | 100%
jpeg-9e | 235 KB | ############################################################## | 100%
cuda-runtime-11.8.0 | 1 KB | ############################################################## | 100%
Preparing transaction: done
Verifying transaction: done
Executing transaction: done
1 Like
Oh, thanks for link!
It worked for me as well,
I appreciate it!
I apologize but, Is there anyway to install torch for cuda-11.8 but using pip?
If there is none, I may have to downgrade cuda to 11.6.
Thank You.
Ubuntu 20.04
Cuda-11.8
Python-3.9
Nvidia-SMI: 525.85.12
ptrblck
February 23, 2023, 7:01am
7
Yes, you can install the nightly PyTorch pip wheels with CUDA 11.8 as seen here .
1 Like
Let me try it.
Thanks a bunch!