满城网站建设邯郸市民网

这里主要介绍在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也可正常使用！