以安装pytorch1.9.1为例
1、在终端查看cuda版本
nvcc -V
2、查看是否有cudnn
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
3、去pytorch官网查看所需版本的指令(https://pytorch.org/)
conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
其中-c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/是清华镜像源,在pycharm终端安装