Troubles with downloading torch preview

JungHwang: <p>Hi,<br>
I am trying to install pytorch preview version with cuda 11.8 on my conda virtual env,<br>
but it is stuck at downloading pytorch-2.0.0.dev202 at 0%.</p>
<p>Is there anyone else facing this issue?</p>
<p>My environments are following:<br>
OS : ubuntu 22.04<br>
GPUs : 4090 x 4<br>
NVIDIA-SMI : 525.85.05<br>
CUDA :12<br>
PYTHON : 3.8</p>

ptrblck: <p>Could you post the install command you were using so that I could double check the install routine?</p>

JungHwang: <p>Oh, yeah sure. I used the following command, which is the one on the official website.</p>
<pre><code class="lang-auto">conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch-nightly -c nvidia
</code></pre>
<p>I was able to download other packages (e.g. torchivision),<br>
but the pytorch-2.0.0 keeps stuck.</p>

ptrblck: <p>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 <a href="https://anaconda.org/pytorch-nightly/pytorch/files">here</a> and <code>conda install</code> it from the local file.<br>
I’ve just ran your provided command and while it also took some time to download the packages, it worked properly for me:</p>
<pre><code class="lang-python">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
</code></pre>

JungHwang: <p>Oh, thanks for link!<br>
It worked for me as well,<br>
I appreciate it!</p>

Raj_Upadhyaya: <p>I apologize but, Is there anyway to install torch for cuda-11.8 but using pip?<br>
If there is none, I may have to downgrade cuda to 11.6.</p>
<p>Thank You.</p>
<p>Ubuntu 20.04<br>
Cuda-11.8<br>
Python-3.9<br>
Nvidia-SMI: 525.85.12</p>

ptrblck: <p>Yes, you can install the nightly PyTorch pip wheels with CUDA 11.8 as seen <a href="https://pytorch.org/get-started/locally/">here</a>.</p>

Raj_Upadhyaya: <p>Let me try it.<br>
Thanks a bunch!</p>