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

接着,输入如下代码

py 复制代码
# main.py
import torch
print(torch.cuda.is_available())

然后点击一下上面的三角形按钮执行一下

返回Ture,torch库及成功了cuda也可正常使用!

相关推荐
递归不收敛10 小时前
吴恩达机器学习课程(PyTorch适配)学习笔记:1.3 特征工程与模型优化
pytorch·学习·机器学习
让子弹飞0211 小时前
永久解决ubuntu网络连接问题
linux·运维·ubuntu
小蕾Java11 小时前
PyCharm 软件使用各种问题 ,解决教程
ide·python·pycharm
星期天要睡觉13 小时前
模型部署——Flask 部署 PyTorch 模型
pytorch·python·flask
西柚小萌新13 小时前
【深入浅出PyTorch】--4.PyTorch基础实战
人工智能·pytorch·python
小虎鲸0015 小时前
PyTorch的安装与使用
人工智能·pytorch·python·深度学习
月墨江山17 小时前
Ubuntu 20.04 使用 Issac Gym 进行宇树G1人形机器人进行强化学习训练(Linux仿真)
linux·ubuntu·机器人
小二·18 小时前
mac下解压jar包
ide·python·pycharm
weixin_4334176719 小时前
PyTorch&TensorFlow
人工智能·pytorch·tensorflow
QMCY_jason19 小时前
ubuntu 24.04 FFmpeg编译 带Nvidia 加速记录
linux·ubuntu·ffmpeg