Windows 安装 PyTorch - Sakaizd's Blog

168 字

1 分钟

Windows 安装 PyTorch

2022-08-16

PyTorch

更换 conda 源为清华源#

1
channels:
2
  - defaults
3
show_channel_urls: true
4
default_channels:
5
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
6
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
7
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
8
custom_channels:
9
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
10
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
11
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
12
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
13
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
14
  pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
15
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

安装 CUDA 和 CUDNN#

CUDA 下载页面 https://developer.nvidia.com/cuda-downloads

nvcc -V查看安装是否成功

cuDNN 下载页面：https://developer.nvidia.com/rdp/cudnn-download

打开 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7 将 cuDNN 里的 bin、include、lib 复制到对应 bin、include、lib

Pytorch#

下载 pytorch GPU 版#

在此下载： https://download.pytorch.org/whl/torch_stable.html

找到对应的 CUDA 版本和 python 版本例如 torch-1.12.1+cu116-cp39-cp39-win_amd64.whl 对应 CUDA 1.16 python3.9 还需要对应 pytorch 官网下载 torchaudio torchvision。

安装 pytorch#

进入下载目录 pip install ./torch-1.12.1+cu116-cp39-cp39-win_amd64.whl

验证#

1
import torch
2

3
x = torch.rand(5, 3)
4
print(x)
5

6
print("torch version ", torch.__version__)
7

8
print("torch.cuda.is_available()", torch.cuda.is_available())
9

10
output:
11
tensor([[0.4556, 0.2505, 0.7284],
12
        [0.4068, 0.0266, 0.3300],
13
        [0.7225, 0.0090, 0.7773],
14
        [0.2523, 0.9489, 0.8674],
15
        [0.4931, 0.7107, 0.3855]])
16
torch version  1.12.1+cu116
17
torch.cuda.is_available() True