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解释器安装的所有的第三方库
- 出现该行即安装成功