安装cuda 、torch、torchvision

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

cuda版本号查询

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

python、torch和cuda对应版本号

然后就可以安装torchvision了:

python、torch和torchvision对应版本号

相关推荐
Thanks_ks6 天前
【第 002 讲】Python 标准开发环境搭建:运行环境 | 环境变量 | IDE 部署 | 配置优化
ide·python·pycharm·开发工具·环境配置·环境变量·编程基础
SkyXZ11 天前
Mac上使用VScode优雅开发STM32
环境配置
不会写程序的未来程序员1 个月前
nvm 安装教程:Node.js 版本管理全攻略 (Win/Mac/Linux) + .nvmrc 实战
linux·macos·node.js·前端开发·环境配置·nvm
陈嘿萌2 个月前
4090显卡配置官方SegFormer运行环境(包含预训练权重下载、环境配置、代码修改、运行 | 验证 | 测试)
环境配置·模型训练·segfomer·预训练权重下载
脱脱克克2 个月前
OpenClaw 安装与配置全教程(Ubuntu 全新虚拟机版)
ubuntu·环境配置·openclaw
_OP_CHEN3 个月前
【MySQL数据库基础】(一)保姆级 MySQL 环境配置教程!CentOS 7+Ubuntu 双系统全覆盖
linux·数据库·sql·mysql·ubuntu·centos·环境配置
youcans_3 个月前
【跟我学YOLO】YOLO26(3)安装配置与快速使用
图像处理·yolo·目标检测·计算机视觉·环境配置
charlie1145141913 个月前
RK3568跑Arch Linux全路程指南(以正点原子的RK3568开发板为例子)
linux·嵌入式·rootfs·教程·环境配置·嵌入式linux·工程实践
charlie1145141913 个月前
SSH X11 转发排查与解决指南(Windows + Xming + Ubuntu)
arm开发·windows·笔记·ubuntu·ssh·个人开发·环境配置
charlie1145141913 个月前
RK3568跑Ubuntu 24.04全路程指南(以正点原子的RK3568开发板为例子)
linux·笔记·ubuntu·rootfs·教程·环境配置·rk3568