windows安装pytorch

windows安装pytorch

1. 安装cuda

pytorch官网我要安装1.12.1对应的cuda有三个版本，我选了11.6

去官网安装这个版本的cuda下载链接

安装后打开命令行输入nvcc -V，可以显示版本则安装成功，如果显示nvcc不是外部命令，进入安装文件下的bin目录，在当前文件夹打开命令行输入.\nvcc.exe -V 正确显示版本信息

2. 安装cudann

安装地址

选择对应版本安装
解压后把这三个文件复制到之前cuda的安装目录


安装完毕

3. 安装pytorch

conda中创建新的环境，conda create -n XXX python=3.7
1

进入环境

conda中创建新的环境，conda activate XXX
1

安装pytorch

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge
1

要是安装出错就换pip

pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116
1

测试

命令行输入python进入python环境

import torch
torch.__version__
torch.cuda.is_available()
1
2
3

安装完成