Skip to content

[Docs] Install PyTorch following on GPU platforms command issue #3238

@UtkarshK12

Description

@UtkarshK12

📚 The doc issue

Regarding this:

Image

on using this, it seems the cuda-toolkit is not installed as mentioned in the doc.

conda install pytorch torchvision -c pytorch

Evidence:

conda list before :

# packages in environment at C:\Users\user\miniconda3\envs\testpip:
#
# Name                     Version          Build            Channel
ca-certificates            2025.7.15        haa95532_0
libffi                     3.4.4            hd77b12b_1
openssl                    3.0.17           h35632f6_0
pip                        24.2             py38haa95532_0
python                     3.8.20           h8205438_0
setuptools                 75.1.0           py38haa95532_0
sqlite                     3.50.2           hda9a48d_1
ucrt                       10.0.22621.0     haa95532_0
vc                         14.3             h2df5915_10
vc14_runtime               14.44.35208      h4927774_10
vs2015_runtime             14.44.35208      ha6b5a95_10
wheel                      0.44.0           py38haa95532_0

Running conda list after pip install:

 blas               pkgs/main/win-64::blas-1.0-mkl
  brotli-python      pkgs/main/win-64::brotli-python-1.0.9-py38hd77b12b_8
  certifi            pkgs/main/win-64::certifi-2024.8.30-py38haa95532_0
  charset-normalizer pkgs/main/noarch::charset-normalizer-3.3.2-pyhd3eb1b0_0
  filelock           pkgs/main/win-64::filelock-3.13.1-py38haa95532_0
  freetype           pkgs/main/win-64::freetype-2.13.3-h0620614_0
  giflib             pkgs/main/win-64::giflib-5.2.2-h7edc060_0
  gmpy2              pkgs/main/win-64::gmpy2-2.1.2-py38h7f96b67_0
  idna               pkgs/main/win-64::idna-3.7-py38haa95532_0
  intel-openmp       pkgs/main/win-64::intel-openmp-2023.1.0-h59b6b97_46320
  jinja2             pkgs/main/win-64::jinja2-3.1.4-py38haa95532_0
  jpeg               pkgs/main/win-64::jpeg-9e-h827c3e9_3
  lcms2              pkgs/main/win-64::lcms2-2.16-hb4a4139_0
  lerc               pkgs/main/win-64::lerc-3.0-hd77b12b_0
  libdeflate         pkgs/main/win-64::libdeflate-1.17-h2bbff1b_1
  libjpeg-turbo      pkgs/main/win-64::libjpeg-turbo-2.0.0-h196d8e1_0
  libpng             pkgs/main/win-64::libpng-1.6.39-h8cc25b3_0
  libtiff            pkgs/main/win-64::libtiff-4.5.1-hd77b12b_0
  libuv              pkgs/main/win-64::libuv-1.48.0-h827c3e9_0
  libwebp            pkgs/main/win-64::libwebp-1.3.2-hbc33d0d_0
  libwebp-base       pkgs/main/win-64::libwebp-base-1.3.2-h3d04722_1
  lz4-c              pkgs/main/win-64::lz4-c-1.9.4-h2bbff1b_1
  markupsafe         pkgs/main/win-64::markupsafe-2.1.3-py38h2bbff1b_0
  mkl                pkgs/main/win-64::mkl-2023.1.0-h6b88ed4_46358
  mkl-service        pkgs/main/win-64::mkl-service-2.4.0-py38h2bbff1b_1
  mkl_fft            pkgs/main/win-64::mkl_fft-1.3.8-py38h2bbff1b_0
  mkl_random         pkgs/main/win-64::mkl_random-1.2.4-py38h59b6b97_0
  mpc                pkgs/main/win-64::mpc-1.1.0-h7edee0f_1
  mpfr               pkgs/main/win-64::mpfr-4.0.2-h62dcd97_1
  mpir               pkgs/main/win-64::mpir-3.0.0-hec2e145_1
  mpmath             pkgs/main/win-64::mpmath-1.3.0-py38haa95532_0
  networkx           pkgs/main/win-64::networkx-3.1-py38haa95532_0
  numpy              pkgs/main/win-64::numpy-1.24.3-py38h79a8e48_1
  numpy-base         pkgs/main/win-64::numpy-base-1.24.3-py38h8a87ada_1
  openjpeg           pkgs/main/win-64::openjpeg-2.5.2-hae555c5_0
  pillow             pkgs/main/win-64::pillow-10.4.0-py38h827c3e9_0
  pysocks            pkgs/main/win-64::pysocks-1.7.1-py38haa95532_0
  pytorch            pytorch/win-64::pytorch-2.4.1-py3.8_cpu_0
  pytorch-mutex      pytorch/noarch::pytorch-mutex-1.0-cpu
  pyyaml             pkgs/main/win-64::pyyaml-6.0.2-py38h827c3e9_0
  requests           pkgs/main/win-64::requests-2.32.3-py38haa95532_0
  sympy              pkgs/main/win-64::sympy-1.13.3-py38haa95532_0
  tbb                pkgs/main/win-64::tbb-2021.8.0-h59b6b97_0
  torchvision        pytorch/win-64::torchvision-0.20.0-py38_cpu
  typing_extensions  pkgs/main/win-64::typing_extensions-4.11.0-py38haa95532_0
  urllib3            pkgs/main/win-64::urllib3-2.2.3-py38haa95532_0
  win_inet_pton      pkgs/main/win-64::win_inet_pton-1.1.0-py38haa95532_0
  xz                 pkgs/main/win-64::xz-5.6.4-h4754444_1
  yaml               pkgs/main/win-64::yaml-0.2.5-he774522_0
  zlib               pkgs/main/win-64::zlib-1.2.13-h8cc25b3_1
  zstd               pkgs/main/win-64::zstd-1.5.6-h8880b57_0

Suggest a potential alternative/fix

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

I was able to use this command to succesfully install the cuda related packages

Since I am not very familiar with the python env, please let me know if I should open a PR to resolve this issue.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions