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解释器安装的所有的第三方库
  • 出现该行即安装成功
相关推荐
三只熊猫5 分钟前
一文打通 AI 知识脉络:大语言模型等关键内容详解
人工智能·python
Kyln.Wu14 分钟前
【python实用小脚本-187】Python一键批量改PDF文字:拖进来秒出新文件——再也不用Acrobat来回导
python·pdf·c#
xnglan1 小时前
蓝桥杯手算题和杂题简易做法
数据结构·数据库·c++·python·算法·职场和发展·蓝桥杯
诗书画唱1 小时前
我的学习认知、高效方法与知识积累笔记
笔记·学习
小徐敲java1 小时前
python-pycharm切换python各种版本的环境与安装python各种版本的环境(pypi轮子下载)
开发语言·python·pycharm
天才测试猿2 小时前
接口自动化测试难点:数据库验证解决方案
自动化测试·软件测试·数据库·python·测试工具·职场和发展·接口测试
我们从未走散2 小时前
设计模式学习笔记-----单例模式
java·笔记·学习·单例模式·设计模式
居7然2 小时前
解锁AI大模型:Prompt工程全面解析
人工智能·prompt·提示词
java1234_小锋5 小时前
[免费]基于Python的影视数据可视化分析系统(Flask+echarts)【论文+源码+SQL脚本】
python·信息可视化·flask·python影视分析·python电影分析
思通数据6 小时前
AI视频监控:重构安防行业智能化新生态
人工智能·安全·目标检测·机器学习·计算机视觉·重构·数据挖掘