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
进入环境
conda中创建新的环境,conda activate XXX
安装pytorch
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge
要是安装出错就换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
测试
命令行输入python进入python环境
import torch
torch.__version__
torch.cuda.is_available()
安装完成