安装cuda 、torch、torchvision

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

cuda版本号查询

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

python、torch和cuda对应版本号

然后就可以安装torchvision了:

python、torch和torchvision对应版本号

相关推荐
大飞记Python1 天前
从零配置Python测试环境:详解路径、依赖与虚拟环境最佳实践
开发语言·python·环境配置·安装目录
闲人编程2 天前
环境配置管理与敏感信息保护
大数据·生命周期·环境配置·加密算法·codecapsule·敏感信息保护
CappuccinoRose7 天前
Docker配置过程完整梳理
后端·python·docker·容器·环境配置
大佬,救命!!!13 天前
C++本地配置OpenCV
开发语言·c++·opencv·学习笔记·环境配置
星云数灵16 天前
使用Anaconda管理Python环境:安装与验证Pandas、NumPy、Matplotlib
开发语言·python·数据分析·pandas·教程·环境配置·anaconda
大佬,救命!!!1 个月前
最新的python3.14版本下仿真环境配置深度学习机器学习相关
开发语言·人工智能·python·深度学习·机器学习·学习笔记·环境配置
琉璃色的星辉2 个月前
Flink-2.0.0在配置文件中修改.pid文件存储位置及其他默认参数
大数据·flink·环境配置·修改参数
Teletele-Lin3 个月前
Miniconda安装与VSCode搭建远程Python、Jupyter开发环境
vscode·python·jupyter·环境配置·远程开发
王伯安呢4 个月前
Python 环境配置初学者指南:从安装到 Pycharm 项目配置
开发语言·python·pycharm·环境配置·初学者
马甲是掉不了一点的<.<4 个月前
Stable Diffusion 环境配置详细指南
stable diffusion·环境配置