pytorch环境配置

这周开始尝试一点小例子,首先开始搭建环境,这里对一些东西做一些记录,以防自己以后还会遇到。

01 pytorch GPU 环境配置

我使用的是pycharm,直接在官网有免费安装包。

继续下载anaconda,使用prompt进行下载库操作,我个人感觉anaconda下载的库会更容易被pycharm检测到。

使用* conda install python* 下载python后,一般用CPU跑数据就可以开始用pycharm写代码了。

如果需要用torch,进行如下操作:

bash 复制代码
#在prompt输入下面语句,进行环境创建
conda create -n pytorch python=3.12

# 检查list里面有没有torch
pip list

#激活环境
conda activate pytorch

查看显卡型号和版本,使用控制面板,即可出现版本信息

进入pytorch首页,下拉可看到推荐版本:

但在下载之前,需要查看对应cuda 的版本,自己的英伟达显卡是否支持,如果显卡版本不够就将GPU 显卡进行版本升级。

直接在英伟达官网下载显卡自动升级的软件,这里最好关闭杀毒软件,不然容易失败,尤其是火绒。下载好后,让版本升级即可。

接着使用官网提供的指令即可,但是我想使用conda安装,最后搜索到相关指令是:

bash 复制代码
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

现在就等待安装好。

接着在pycharm里面添加解释器,需要先新建文件夹,新建解释器,我的选择路径如下:

接着选中正确的pytorch的环境即可,如果出现红框,环境那一栏,先点生成新的,再回到选择现有的,红框就消失了。

接着输入语句检验是否成功。

我直接使用的pycharm 的控制台,返回得到true,说明环境配置成功。

02 jupyter notebook中配置torch

依旧是用conda进行操作。

先用conda activate pytorch 进入pytorch环境,查看有没有所需要的package(箭头位置的包)

发现pytorch环境没有,使用

bash 复制代码
conda install nb_conda

进行对应包的安装,接着输入jupyter notebook,即可进入,使用new创建新文件,进行检验,就这样成功进行了环境配置

相关推荐
张较瘦_21 分钟前
[论文阅读] 人工智能 + 软件工程 | Trae Agent:让AI智能体高效解决仓库级软件问题,登顶SWE-bench排行榜
论文阅读·人工智能·软件工程
啊哈哈哈哈哈啊哈哈21 分钟前
G9打卡——ACGAN
python·生成对抗网络·gan
ALLSectorSorft23 分钟前
相亲小程序用户注册与登录系统模块搭建
java·大数据·服务器·数据库·python
nuoxin11432 分钟前
CY7C68013A-56LTXC -USB2.0控制器芯片-富利威,国产CBM9002A-56ILG可替代
网络·人工智能·单片机·嵌入式硬件·硬件工程
落叶阳光40 分钟前
2025.8-12月 AI相关国内会议
人工智能·会议
小一亿42 分钟前
【0基础PS】Photoshop (PS) 理论知识
学习·平面·adobe·信息可视化·photoshop
FL16238631291 小时前
红花识别分割数据集labelme格式144张1类别
人工智能·深度学习
程序员JerrySUN1 小时前
OpenCV 全解读:核心、源码结构与图像/视频渲染能力深度对比
linux·人工智能·驱动开发·opencv·计算机视觉·缓存·音视频
慕y2741 小时前
Java学习第九十一部分——OkHttp
java·开发语言·学习
caisexi1 小时前
Windows批量启动java服务bat脚本
java·windows·python