在windows 安装好anaconda和对应版本的pytorch、torchvision后,检查是否安装成功
(具体可参考:[Linux ]安装Anaconda、python、验证cuda是否可用)
(anaconda和python版本对应关系)
如果想在pycharm中使用torch 环境
开Anaconda Prompt,输入指令
conda create -n pytorch python=3.7
新建一个环境,环境名字叫pytorch
输入y,回车安装即可
PyCharm关联
打开PyCharm,点击文件 —>设置
测试Pytorch是否安装成功
在任意文件中输入以下代码并运行:
import torch
print(torch.version)
print(torch.cuda.is_available())