安装cuda 、torch、torchvision

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

cuda版本号查询

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

python、torch和cuda对应版本号

然后就可以安装torchvision了:

python、torch和torchvision对应版本号

相关推荐
陈嘿萌8 天前
4090显卡配置官方SegFormer运行环境(包含预训练权重下载、环境配置、代码修改、运行 | 验证 | 测试)
环境配置·模型训练·segfomer·预训练权重下载
脱脱克克14 天前
OpenClaw 安装与配置全教程(Ubuntu 全新虚拟机版)
ubuntu·环境配置·openclaw
_OP_CHEN1 个月前
【MySQL数据库基础】(一)保姆级 MySQL 环境配置教程!CentOS 7+Ubuntu 双系统全覆盖
linux·数据库·sql·mysql·ubuntu·centos·环境配置
youcans_1 个月前
【跟我学YOLO】YOLO26(3)安装配置与快速使用
图像处理·yolo·目标检测·计算机视觉·环境配置
charlie1145141911 个月前
RK3568跑Arch Linux全路程指南(以正点原子的RK3568开发板为例子)
linux·嵌入式·rootfs·教程·环境配置·嵌入式linux·工程实践
charlie1145141911 个月前
SSH X11 转发排查与解决指南(Windows + Xming + Ubuntu)
arm开发·windows·笔记·ubuntu·ssh·个人开发·环境配置
charlie1145141911 个月前
RK3568跑Ubuntu 24.04全路程指南(以正点原子的RK3568开发板为例子)
linux·笔记·ubuntu·rootfs·教程·环境配置·rk3568
腾飞开源2 个月前
03_Dify 使用 Docker Compose 部署
环境配置·docker compose·dify·docker部署·容器化部署·ai开发平台·dify开源平台
●VON2 个月前
Flutter for OpenHarmony 21天训练营 Day01 总结:从零搭建开发环境
flutter·环境配置·openharmony·训练营·跨平台开发·von
wrj的博客2 个月前
python环境安装
python·学习·环境配置