Pytorch配置（2）：在虚拟环境中安装pytorch

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解释器即可