安装cuda 、torch、torchvision

整体思路为 先根据 nvidia-smi 指令 看看自己的cuda版本,然后安装对应的cuda驱动 对应关系表如下:

cuda版本号查询

接着,根据安装的cuda以及python版本,安装对应的torch,cuda、python与torch对应版本号如下:

python、torch和cuda对应版本号

然后就可以安装torchvision了:

python、torch和torchvision对应版本号

相关推荐
SkyXZ11 天前
Linux下Docker及Nvidia Container ToolKit安装教程
环境配置
落日弥漫的橘_21 天前
Node.js下载安装及环境配置教程 (详细版)
前端·node.js·环境配置·node安装教程
laugh123211 个月前
性能飞跃!TensorRT-YOLO 6.0 全面升级解析与实战指南
环境配置·tensorrt·cuda·c/c++
颜淡慕潇1 个月前
【K8S系列】在 K8S 中使用 Values 文件定制不同环境下的应用配置
云原生·容器·kubernetes·环境配置
SkyXZ1 个月前
VScode配置X11转发!让你彻底摆脱显示屏!!!
环境配置
SkyXZ1 个月前
告别虚拟机!WSL2安装配置教程!!!
环境配置
SkyXZ1 个月前
老奶奶看了都会的WSL2连接USB设备教程!
环境配置
卡戎-caryon2 个月前
【MySQL】01.MySQL环境安装
数据库·笔记·mysql·环境配置
SkyXZ2 个月前
2025年了,你还不会配置Jetson Orin NX嘛?
环境配置·jetson
程序猿000001号2 个月前
Conda 使用全解析:从入门到精通
conda·环境配置