仅在这里介绍一下并记录自己安装pytorch的过程,方便后续换机时快速恢复学习环境。

PS C:\Users\Desktop> python
Python 3.9.12 (main, Apr  4 2022, 05:22:27) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32

Warning:
This Python interpreter is in a conda environment, but the environment has
not been activated.  Libraries may fail to load.  To activate this environment
please see https://conda.io/activation

Type "help", "copyright", "credits" or "license" for more information.
>>> import pytorch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'pytorch'
>>> exit()

cuda11.7, CPU版本的pytorch

PS C:\Users\Desktop> conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
Collecting package metadata (current_repodata.json): done
Solving environment: \
The environment is inconsistent, please check the package plan carefully
The following packages are causing the inconsistency:

  - defaults/win-64::anaconda==2022.05=py39_0
  - defaults/win-64::astropy==5.0.4=py39h080aedc_0
  - ......
  
failed with initial frozen solve. Retrying with flexible solve.
Solving environment: failed with repodata from current_repodata.json, will retry with next repodata source.
Collecting package metadata (repodata.json): done
Solving environment: \
The environment is inconsistent, please check the package plan carefully
The following packages are causing the inconsistency:

  - defaults/win-64::anaconda==2022.05=py39_0
  - defaults/win-64::astropy==5.0.4=py39h080aedc_0
  - ......
done

安装包与目录(省略版)

## Package Plan ##

  environment location: C:\Program Files\Anaconda
  added / updated specs:
    - pytorch
    - pytorch-cuda=11.7
    - torchaudio
    - torchvision
```python
The following packages will be downloaded:

    package                    |            build
    ---------------------------|-----------------
    _anaconda_depends-2022.05  |           py39_0           7 KB
    anaconda-custom            |           py39_1           4 KB
    certifi-2022.9.24          |   py39haa95532_0         154 KB
    conda-22.11.1              |   py39haa95532_4         892 KB
    cuda-11.7.1                |                0           1 KB  nvidia
    openssl-1.1.1s             |       h2bbff1b_0         5.5 MB
    pytorch-1.13.1             |py3.9_cuda11.7_cudnn8_0        1.10 GB  pytorch
    pytorch-cuda-11.7          |       h67b0de4_1           3 KB  pytorch
    ruamel.yaml-0.17.21        |   py39h2bbff1b_0         174 KB
    ruamel.yaml.clib-0.2.6     |   py39h2bbff1b_1         101 KB
    torchaudio-0.13.1          |       py39_cu117         4.6 MB  pytorch
    torchvision-0.14.1         |       py39_cu117         7.5 MB  pytorch
    ------------------------------------------------------------
                                           Total:        2.93 GB
The following NEW packages will be INSTALLED:

  _anaconda_depends  pkgs/main/win-64::_anaconda_depends-2022.05-py39_0 None
  cuda               nvidia/win-64::cuda-11.7.1-0 None
  cuda-cccl          nvidia/win-64::cuda-cccl-11.7.91-0 None
  cuda-command-line~ nvidia/win-64::cuda-command-line-tools-11.7.1-0 None
  cuda-compiler      nvidia/win-64::cuda-compiler-11.7.1-0 None
  cuda-cudart        nvidia/win-64::cuda-cudart-11.7.99-0 None
  cuda-cudart-dev    nvidia/win-64::cuda-cudart-dev-11.7.99-0 None
  cuda-cuobjdump     nvidia/win-64::cuda-cuobjdump-11.7.91-0 None
  cuda-cupti         nvidia/win-64::cuda-cupti-11.7.101-0 None
  cuda-cuxxfilt      nvidia/win-64::cuda-cuxxfilt-11.7.91-0 None
  cuda-demo-suite    nvidia/win-64::cuda-demo-suite-12.0.76-0 None
  cuda-documentation nvidia/win-64::cuda-documentation-12.0.76-0 None
  cuda-libraries     nvidia/win-64::cuda-libraries-11.7.1-0 None
  cuda-libraries-dev nvidia/win-64::cuda-libraries-dev-11.7.1-0 None
  cuda-memcheck      nvidia/win-64::cuda-memcheck-11.8.86-0 None
  cuda-nsight-compu~ nvidia/win-64::cuda-nsight-compute-12.0.0-0 None
  cuda-nvcc          nvidia/win-64::cuda-nvcc-11.7.99-0 None
  cuda-nvdisasm      nvidia/win-64::cuda-nvdisasm-12.0.76-0 None
  cuda-nvml-dev      nvidia/win-64::cuda-nvml-dev-11.7.91-0 None
  cuda-nvprof        nvidia/win-64::cuda-nvprof-12.0.90-0 None
  cuda-nvprune       nvidia/win-64::cuda-nvprune-11.7.91-0 None
  cuda-nvrtc         nvidia/win-64::cuda-nvrtc-11.7.99-0 None
  cuda-nvrtc-dev     nvidia/win-64::cuda-nvrtc-dev-11.7.99-0 None
  cuda-nvtx          nvidia/win-64::cuda-nvtx-11.7.91-0 None
  cuda-nvvp          nvidia/win-64::cuda-nvvp-12.0.90-0 None
  cuda-runtime       nvidia/win-64::cuda-runtime-11.7.1-0 None
  cuda-sanitizer-api nvidia/win-64::cuda-sanitizer-api-12.0.90-0 None
  cuda-toolkit       nvidia/win-64::cuda-toolkit-11.7.1-0 None
  cuda-tools         nvidia/win-64::cuda-tools-11.7.1-0 None
  cuda-visual-tools  nvidia/win-64::cuda-visual-tools-11.7.1-0 None
  libcublas          nvidia/win-64::libcublas-11.10.3.66-0 None
  libcublas-dev      nvidia/win-64::libcublas-dev-11.10.3.66-0 None
  libcufft           nvidia/win-64::libcufft-10.7.2.124-0 None
  libcufft-dev       nvidia/win-64::libcufft-dev-10.7.2.124-0 None
  libcurand          nvidia/win-64::libcurand-10.3.1.50-0 None
  libcurand-dev      nvidia/win-64::libcurand-dev-10.3.1.50-0 None
  libcusolver        nvidia/win-64::libcusolver-11.4.0.1-0 None
  libcusolver-dev    nvidia/win-64::libcusolver-dev-11.4.0.1-0 None
  libcusparse        nvidia/win-64::libcusparse-11.7.4.91-0 None
  libcusparse-dev    nvidia/win-64::libcusparse-dev-11.7.4.91-0 None
  libnpp             nvidia/win-64::libnpp-11.7.4.75-0 None
  libnpp-dev         nvidia/win-64::libnpp-dev-11.7.4.75-0 None
  libnvjpeg          nvidia/win-64::libnvjpeg-11.8.0.2-0 None
  libnvjpeg-dev      nvidia/win-64::libnvjpeg-dev-11.8.0.2-0 None
  nsight-compute     nvidia/win-64::nsight-compute-2022.4.0.15-0 None
  numpy-base         pkgs/main/win-64::numpy-base-1.21.5-py39hca35cd5_1 None
  pytorch-cuda       pytorch/noarch::pytorch-cuda-11.7-h67b0de4_1 None
  ruamel.yaml        pkgs/main/win-64::ruamel.yaml-0.17.21-py39h2bbff1b_0 None
  ruamel.yaml.clib   pkgs/main/win-64::ruamel.yaml.clib-0.2.6-py39h2bbff1b_1 None

The following packages will be UPDATED:

  ca-certificates                      2022.3.29-haa95532_1 --> 2022.10.11-haa95532_0 None
  certifi                          2021.10.8-py39haa95532_2 --> 2022.9.24-py39haa95532_0 None
  conda                               22.9.0-py39haa95532_0 --> 22.11.1-py39haa95532_4 None
  openssl                                 1.1.1n-h2bbff1b_0 --> 1.1.1s-h2bbff1b_0 None
  pytorch                    1.12.1-py3.9_cuda11.3_cudnn8_0 --> 1.13.1-py3.9_cuda11.7_cudnn8_0 None
  torchaudio                              0.12.1-py39_cu113 --> 0.13.1-py39_cu117 None
  torchvision                             0.13.1-py39_cu113 --> 0.14.1-py39_cu117 None

The following packages will be DOWNGRADED:

  anaconda                                   2022.05-py39_0 --> custom-py39_1 None
Proceed ([y]/n)? y

安装进度:

Downloading and Extracting Packages
cuda-tools-11.7.1    | 1 KB      | ############################################################################ | 100%
libnpp-11.7.4.75     | 294 KB    | ############################################################################ | 100%
libcusolver-11.4.0.1 | 29 KB     | ############################################################################ | 100%
Preparing transaction: done
Verifying transaction: done
Executing transaction: done
Retrieving notices: ...working... done
测试安装成功。
```python
PS C:\Users\Hkl\Desktop> python
Python 3.9.12 (main, Apr  4 2022, 05:22:27) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32

Warning:
This Python interpreter is in a conda environment, but the environment has
not been activated.  Libraries may fail to load.  To activate this environment
please see https://conda.io/activation

Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐