安装cuda 、torch、torchvision

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

cuda版本号查询

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

python、torch和cuda对应版本号

然后就可以安装torchvision了:

python、torch和torchvision对应版本号

相关推荐
SpikeKing1 个月前
LLM - 使用 LLaMA-Factory 微调大模型 环境配置与训练推理 教程 (1)
人工智能·llm·大语言模型·llama·环境配置·llamafactory·训练框架
炼丹师小米2 个月前
Ubuntu24.04.1系统下VideoMamba环境配置
python·环境配置·videomamba
坚果仙人3 个月前
【环境配置】AST: Asymmetric Student-Teacher Networks for Industrial Anomaly Detection
环境配置
QH_ShareHub3 个月前
解决R包依赖版本不兼容问题
r语言·环境配置·r包
DieSnowK3 个月前
[Redis][环境配置]详细讲解
数据库·redis·分布式·缓存·环境配置·新手向·详细讲解
FORGIVEN_H3 个月前
编程环境常用命令合集
环境配置
Spy973 个月前
【最新2024】Windows系统,C++ 操作 MySQL 数据库
数据库·c++·windows·mysql·环境配置·mysql-connector
nodototao3 个月前
yolov8 save_dir设置setting文件
yolo·环境配置
SkyXZ3 个月前
Mac上HomeBrew安装及换源教程
mac·环境配置