2024.9最新:CUDA安装,pytorch库安装

目录

一、CUDA安装

1.查看自己电脑适配的CUDA的最高版本

2.安装CUDA

3.检查环境变量是否配置,安装是否成功

二、pytorch库安装

1.pytorch库下载

2.选择合适的版本

3.查看版本


一、CUDA安装

1.查看自己电脑适配的CUDA的最高版本

  • 在命令提示符里输入nvidia-smi
  • 表格右上角显示的CUDA版本是该电脑适配的最高版本
  • 一般下载比该版本低一点的版本,因为会更稳定

2.安装CUDA

CUDA下载地址:

CUDA Toolkit Archive | NVIDIA Developer

  • 自选版本下载,我下载的是11.8.0版本的

  • 选择windows,x86,win10,一般选择local本地下载

  • 推荐使用迅雷下载,浏览器下载很慢

  • 选择默认路径即可

  • 正在安装

  • 自动检查系统兼容性

  • 选择自定义安装

  • 第一次安装全选

  • 使用默认位置即可,放c盘就行,尽量不要放d,e盘

  • 下一步,直到安装完成

3.检查环境变量是否配置,安装是否成功

  • 可以在命令提示符输入set cuda 来查看环境变量是否配置

  • 也可以在系统变量里查看有无上面的路径

  • 最后在命令提示符输入nvcc -V 或者 nvcc --version查看安装好的CUDA版本及其信息

二、pytorch库安装

1.pytorch库下载

  • 复制下方红框里的网址

  • 进入网页之后下滑找到torch,点击进入

2.选择合适的版本

  • 按ctrl+f搜索网页内容,
  • 找到符合条件的torch版本进行安装
    • cp表示python解释器的版本,需要与自己所用的解释器版本一致
    • cu表示CUDA版本,可以比自己下载的CUDA版本低
    • torch选择1.0的版本,不要选择2.0及以后的版本

3.查看版本

  • 安装好了之后在命令提示符里输入pip list 即可查看python解释器安装的所有的第三方库
  • 出现该行即安装成功
相关推荐
100个铜锣烧4 小时前
高级提示技术:Chain-of-Thought与ReAct——让大模型学会“思考”和“行动”
人工智能·大模型·提示词工程
JackHCC4 小时前
快手OneRetrieval:可编辑生成式电商召回
人工智能·机器学习
hhzz4 小时前
基于监控视频的水位尺自动识别技术方案与实现
python·opencv·yolo·图像识别·cv
yongche_shi4 小时前
ragas官方文档中文版(五十)
开发语言·python·ai·ragas·如何评估和改进 rag 应用
前端之虎陈随易4 小时前
编程语言级别的Skill市场,AI Agent 的未来形态
前端·vue.js·人工智能·typescript·node.js
QiLinkOS4 小时前
第三视觉理解徐玉生与他的商业活动(30)
大数据·c++·人工智能·算法·开源协议
武汉唯众智创5 小时前
当汉字成为心理CT:AI汉字联想投射分析的技术实现与心理评估价值
人工智能·ai心理健康·ai心理评估·本土化心理测评·校园心理健康解决方案·ai心理监测·多模态情绪模型
Longvox5 小时前
Agent为什么会死循环?
人工智能·ai编程
陈天伟教授5 小时前
FreeCAD 启动后小窗口闪现即退的解决思路
人工智能·机器人·工业设计
weixin_408099675 小时前
OCR批量识别图片方案:从手动处理到自动化API系统(Python/Java/PHP实战)
图像处理·python·ocr·文字识别·api调用·批量识别·石榴智能