1、在nvidia官网安装显卡驱动的最新版本
2、cmd:nvidia-smi,得到cuda driver的版本,如图cuda driver的版本为12.23、打开Pytorch官网,确定CUDA Runtime版本为11.7(runtime版本最好小于driver版本)
方法1:从官网(PyTorch)采用命令行下载
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
安装cudatoolkit:conda install cudatoolkit=11.7 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
方法2:添加镜像源
安装pytorch torchvision等包:
conda install pytorch torchvision torchaudio -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
安装cudatoolkit(11.7不行,改成11.3了):conda install cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
方法3:cuda为11.7版本可用
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
4、卸载(用不到)
conda uninstall pytorch torchvision torchaudio cudatoolkit
5、pycharm集成
将虚拟环境添加至pycharm的python解释器即可