深度学习DL 之 安装PyTorch·GPU版、CUDA(本人Anaconda、Python、PyCharm已提前安装好)

1. 安装CUDA (CUDA的核心作用是:解锁 GPU 的并行计算能力,用于处理非图形任务)

根据如下链接,确定 CUDA 版本进行下载安装!!!

https://mp.weixin.qq.com/s/0v8jVp9dOq_8jU4SubDgDQ

安装完成后,查看CUDA版本:Win+R后输入cmd,进入命令提示符,输入nvcc -V,注意中间有空格!!!(如图 已成功)

配置环境

默认路径下 如下四个文件

2. 安装Python虚拟环境,并在虚拟环境中安装对应版本的pytorch

在conda上装一个虚拟环境,Python版本也选一个与电脑cuda对应的,这里我创建的3.9版本

(可参考: https://blog.csdn.net/YYWWQQ5656/article/details/153411619?spm=1001.2014.3001.5501)

下面是我问的豆包和一个博主推荐的对应版本表格

我的显卡信息:

对应CUDA11.7

下面安装pytorch(我用的保险方法:下载.whl文件并安装,博主说pip直接运行代码容易死机哈哈哈)

相关网址:

https://pytorch.org/get-started/locally/#windows-pip

https://download.pytorch.org/whl/cu117 ++标红部分++ 要换成自己的版本

依次点击,选择自己需要的版本,进行点击下载(https://pytorch.org/get-started/previous-versions/在这里看对应版本)

比如我的 CUDA 版本为:1.13,就对应这三个版本的torchvision等(注意不要选+cpu的!!!)

对应好 这个格式(Ctrl+F 可直接输入关键词检索)

要下载很长时间,漫长的等待~~~

安装好之后,输入 conda list,查看是否包括以下三个库(这里注明一下,如果conda list不显示,可以更换pip list 就可以显示这几个库拉,因为我们是pip安装的)

3.检验torch和cuda是否可用

可以看这个up主:https://www.bilibili.com/video/BV1cD4y1H7Tk/?spm_id_from=333.337.search-card.all.click&vd_source=d582f273a1efcb8a57b6bade9e5ef400,视频下里面有资源哦

相关推荐
deepseek2312 小时前
SQLite WAL 开了仍报 database is locked?Python 后台任务的事务与幂等修复
数据库·python·sqlite·并发编程·wal
杨逢昌工厂6S管理12 小时前
116——杨逢昌:归位三定闭环法的旺季轻量化执行方案与工具清单
经验分享·笔记·职场和发展·学习方法
一滴泪的时间12 小时前
不同行业 RFID 硬件落地场景对比分析
经验分享
计算机编程指导师12 小时前
计算机大数据项目推荐:基于Hadoop+Django的人工智能社交媒体情绪数据分析可视化系统源码 毕业设计 选题推荐 毕设选题 数据分析 机器学习
大数据·hadoop·python·计算机·spark·毕业设计·社交媒体
Aurorar0rua12 小时前
CS50 x 2024 Notes Data structures - 08
c语言·开发语言·学习方法
高级程序源13 小时前
springboot荆州旅游推介平台24175-计算机课程设计、毕业设计
vue.js·spring boot·python·mysql·django·课程设计
企业数字化笔记13 小时前
新租户注册成功却不能用?开通状态、初始化恢复与可用性验收
开发语言·python
Guarding and trust13 小时前
项目拆解(二):模型输出是‘野‘的——三层兜底+重分析,把AI变成可靠零件
java·人工智能·python·大模型·fastapi·ollama·错误兜底
用户2986985301413 小时前
Python 实战:将 HTML 内容保存为图像文件
后端·python·html
for_ever_love__13 小时前
LangChain MultiVectorRetriever:一个文档存多个向量
人工智能·python·自然语言处理·langchain·ai大模型·多向量·摘要检索