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

相关推荐
西柚小萌新37 分钟前
【深入浅出PyTorch】--上采样+下采样
人工智能·pytorch·python
虚拟指尖2 小时前
Ubuntu编译安装COLMAP【实测编译成功】
linux·运维·ubuntu
dlraba8025 小时前
PyTorch 模型部署实战:用 Flask 搭图像分类 API
pytorch·分类·flask
StarPrayers.6 小时前
旅行商问题(TSP)(2)(heuristics.py)(TSP 的两种贪心启发式算法实现)
前端·人工智能·python·算法·pycharm·启发式算法
周之鸥6 小时前
从零部署 Astro 静态网站到云服务器(含 HTTPS 一键配置)
运维·服务器·ubuntu·http·https·astro
Wx-bishekaifayuan7 小时前
基于微信小程序的社区图书共享平台设计与实现 计算机毕业设计源码44991
javascript·vue.js·windows·mysql·pycharm·tomcat·php
西柚小萌新7 小时前
【深入浅出PyTorch】--6.2.PyTorch进阶训练技巧2
人工智能·pytorch·python
paoqi 包奇8 小时前
pycharm中使用anaconda指定虚拟环境
ide·python·pycharm
Kaydeon9 小时前
【AIGC】50倍加速!NVIDIA蒸馏算法rCM:分数正则化连续时间一致性模型的大规模扩散蒸馏
人工智能·pytorch·python·深度学习·计算机视觉·aigc
中草药z10 小时前
【Docker】零基础上手:原理+Ubuntu/Windows GUI 安装 + 镜像源 / 目录优化
运维·ubuntu·docker·容器·gui·安装·cgroups