安装好anaconda和对应版本的pytorch、torchvision后PyCharm关联pytorch

在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())
在这里插入图片描述
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/weixin_49627776/article/details/117871111