这里主要介绍在conda+pytorch都正确安装的前提下,如何通过pycharm建立开发环境;
Ubuntu下Anaconda+PyCharm搭建PyTorch环境
系统环境:Ubuntu22.04
conda: conda 23.11.0
pycharm:如下
conda+pytorch的安装教程介绍,请点击这里:ubuntu下的minconda+pytorch(gpu版本)安装
创建新项目
注意,首选选择Custom environment
,并在随后的Environment
中选择Select existing
,然后Type
设置Conda
;最后重点在最后的Environment
找到并选择你在Conda创建的虚拟环境,如下所示:
pyt-gpu-2.0
这个conda环境为提前创建好的环境,如何创建,可参考这篇文章请点击这里:minconda+pytorch(gpu版本)
测试一下pycharm中的torch库是否正常
项目创建好后,创建一个新文件main.py
接着,输入如下代码
# main.py
import torch
print(torch.cuda.is_available())
然后点击一下上面的三角形按钮执行一下
返回Ture,torch库及成功了cuda也可正常使用!