安装cuda 、torch、torchvision

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

cuda版本号查询

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

python、torch和cuda对应版本号

然后就可以安装torchvision了:

python、torch和torchvision对应版本号

相关推荐
mahuifa11 天前
PySide环境配置及工具使用
python·qt·环境配置·开发经验·pyside
瓢儿菜201817 天前
Proteus8.17-LCD12864液晶屏幕仿真模型
单片机·proteus·环境配置·仿真
妄想成为master2 个月前
如何完美安装GPU版本的torch、torchvision----解决torch安装慢 无法安装 需要翻墙安装 安装的是GPU版本但无法使用的GPU的错误
人工智能·pytorch·python·环境配置
瓢儿菜20183 个月前
proteus8.17 环境配置
单片机·proteus·环境配置
mahuifa3 个月前
pyqt环境配置
python·qt·pycharm·pyqt·环境配置
Roadinforest3 个月前
wsl2配置xv6全解(包括22.04Jammy)
linux·操作系统·qemu·wsl·环境配置·xv6
luoyayun3614 个月前
Trae+Qt+MSVC环境配置
vscode·qt·环境配置·trae qt
SkyXZ5 个月前
Linux下Docker及Nvidia Container ToolKit安装教程
环境配置
落日弥漫的橘_5 个月前
Node.js下载安装及环境配置教程 (详细版)
前端·node.js·环境配置·node安装教程
laugh123215 个月前
性能飞跃!TensorRT-YOLO 6.0 全面升级解析与实战指南
环境配置·tensorrt·cuda·c/c++