python3.9（anaconda）+pytorch（GPU）

1.新建一个python环境

安装anaconda  5.0.1

安装名为TorchGPU的conda虚拟环境

conda create -n TorchGPU python==3.9

最终效果

conda env list

出错

在如下页面添加conda环境后，解释器列表里没反应

• 关于错误的链接
• https://youtrack.jetbrains.com/issue/PY-57996

原因

• conda版本低，建议更新。
• conda和pycharm无法协同工作

解决

找不到就不找了，直接找到安装好的环境就行了

选择system interpreter

2.安装pytorch（GPU版本的）

我的配置：python3.9，显卡NVIDIA GeForce GTX 1050Ti

安装pythorch（GPU版本）的指令

pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu113

3.测试安装效果

#在TorchGPU环境下输入
import torch     #不报错就算成功
torch.cuda.is_available() #输出true就算成功