1.首先打开pycharm 新建项目
2.点击左上角setting,如下图
3.点击左栏的python interpreter
4.选择对应的解释器python interpreter ,点击ok完成
5.如果没有你要的解释器,则点击下面的选择show all
6.再选择+号
7.选择conda 环境,然后选择已经存在的环境 existing environment
8.一路ok回去
9.再main.py文件中写入下面代码测试
import torch
print(torch.cuda.is_available())
x=torch.randn(3,5)
print(x)
true表示pytorch GPU版本可以使用了
pycharm中导入pytorch
最新推荐文章于 2024-06-08 22:24:38 发布