CUDA的安装 — 用自己电脑的GPU跑深度学习模型

本文指导使用Anaconda创建的Python开发环境中安装PyTorch,首先通过nvidia-smi检查CUDA版本,建议选择兼容的CUDA11.7。然后在PyTorch官网获取安装命令,激活虚拟环境后执行该命令。如果遇到pip版本问题,需在虚拟环境中升级pip。
部署运行你感兴趣的模型镜像

本文有用的前提:你是用Anaconda装的python开发环境,用pytorch构建模型。

如何用Anaconda装的python开发环境,请参考另一篇博客,链接如下

 Anaconda + Pycharm,利用Anaconda安装python并配置虚拟环境,包括sklearn、pytorch的安装_learninger_lt7的博客-优快云博客

一、Win + R, cmd —> 输入

nvidia-smi

找到

这里的CUDA Version指的是你的电脑最高支持的CUDA版本,它是向下兼容的,因此,低于12.0版本的CUDA都可安装。

二、进入pytorch官网,PyTorch

往下滑,找到这个,按我这么选

建议选择CUDA11.7版本,我试过11.8,莫名的报错。 

 复制灰色部分的代码。

三、打开开始菜单,找到Anaconda Powershell Prompt,点击进入

 输入:conda activate myEnvs_3_7_16  # (红色部分应根据你自己的虚拟环境名填写)

成功激活虚拟环境后,接着在红线处粘贴刚刚复制的那段代码,然后回车等待安装即可。

如果提示pip版本问题,就在当前界面(虚拟环境已激活的情况下)升级pip,升级pip的代码如下:

python -m pip install --upgrade pip

您可能感兴趣的与本文相关的镜像

Python3.9

Python3.9

Conda
Python

Python 是一种高级、解释型、通用的编程语言,以其简洁易读的语法而闻名,适用于广泛的应用,包括Web开发、数据分析、人工智能和自动化脚本

评论 1
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

learninger_lt7

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值