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也可正常使用!

相关推荐
☼←安于亥时→❦8 小时前
PyTorch 梯度与微积分
人工智能·pytorch·python
HetFrame11 小时前
John the Ripper jumbo + HashCat 破解压缩密码 ubuntu amd GPU
linux·ubuntu·amd·密码破解·john·压缩密码·hashcat
saynaihe14 小时前
关于Ubuntu的 update造成的内核升级
linux·运维·服务器·ubuntu·devops
缘友一世16 小时前
PyTorch深度学习实战【10】之神经网络的损失函数
pytorch·深度学习·神经网络
深耕AI16 小时前
【参数详解与使用指南】PyTorch MNIST数据集加载
人工智能·pytorch·python
星期天要睡觉17 小时前
深度学习——基于 PyTorch 的 CBOW 模型实现自然语言处理
pytorch·深度学习·自然语言处理
小叮当⇔17 小时前
PYcharm——pyqt音乐播放器
ide·pycharm·pyqt
lianghaibos17 小时前
虚拟环境下,pythonDjango项目配置pycharm运行/debugger运行
ide·python·pycharm
楼田莉子17 小时前
Python学习——字典和文件
开发语言·python·学习·pycharm
胡乱编胡乱赢18 小时前
关于在pycharm终端连接服务器
人工智能·深度学习·pycharm·终端连接服务器