Ubuntu下Anaconda+PyCharm搭建PyTorch环境

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

  • 6
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值