Troubles with downloading torch preview

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

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.

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

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!