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创建新文件,进行检验,就这样成功进行了环境配置

相关推荐
MiaoChuAI2 分钟前
豆包AI PPT与秒出PPT对比评测:谁更适合你?
人工智能·powerpoint
hweiyu002 分钟前
Python办公之Excel(openpyxl)、PPT(python-pptx)、Word(python-docx)
python·powerpoint·excel
java技术之路13 分钟前
【免费AI文档助手开发实战系列】基于正则表达式的PDF脱敏python服务构建(一)
python·pdf·pymupdf·免费pdf脱敏
waynaqua15 分钟前
FastAPI开发AI应用五:模型新增图片理解
python·openai·fastapi
waynaqua15 分钟前
FastAPI开发AI应用四:新增豆包、kimi模型
python·fastapi·ai编程
%KT%20 分钟前
简单聊聊多模态大语言模型MLLM
人工智能·语言模型·自然语言处理
唐某人丶28 分钟前
教你如何用 JS 实现一个 Agent 系统(1)—— 认识 Agentic System
前端·人工智能
泡泡茶壶_ovo35 分钟前
RORPCAP: retrieval-based objects and relations prompt for image captioning
人工智能·深度学习·计算机视觉·语言模型·prompt·多模态·imagecaptioning
MaxCode-139 分钟前
单智能体篇:Prompt工程艺术
大数据·人工智能·prompt
小鹿的工作手帐1 小时前
有鹿机器人:智慧清洁新时代的引领者
人工智能·科技·机器人