服务器上Cuda+Pytorch兼容性的那些问题

服务器上如何搭建匹配版本的cuda+pytorch:

1.查询nvidia中驱动版本与cuda最高兼容版本

命令行输入:nvidia-smi

2.选择对应的python

Driver Version与NVIDIA-SMI决定当前服务器上能兼容最高cuda和Pytorch版本

查询匹配方式:1.官网查询 2.问AI(可能存在不准确情况)

CUDA版本与显卡驱动版本对照表(更新至2022.10.26 - CUDA11.8)

3.cuda+torch离线包(网站附torchvision离线包):

http://download.pytorch.org/whl/torch_stable.html

4.torch与torchvision版本对应:

PyTorch中torch、torchvision、torchaudio、torchtext版本对应关系_torch2.0.1对应的torchvision-CSDN博客(经尝试可能存在一定问题)

其它方法:问AI对应

bash 复制代码
import torch # 如果pytorch安装成功即可导入  
print(torch.cuda.is_available()) # 查看CUDA是否可用  
print(torch.cuda.device_count()) # 查看可用的CUDA数量  
print(torch.version.cuda) # 查看CUDA的版本号

cuda+pytorch以及服务器驱动不匹配的那些问题:

1.RuntimeError: The NVIDIA driver on your system is too old.-CSDN博客

2.RuntimeError:No such operator torchvision::nms

3.UserWarning: NVIDIA GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
If you want to use the NVIDIA GeForce RTX 3080 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

相关推荐
男孩李12 分钟前
浅谈代理流程自动化 (APA)
运维·人工智能·自动化
尚久龙24 分钟前
安卓学习 之 用户登录界面的简单实现
android·运维·服务器·学习·手机·android studio·安卓
high20111 小时前
【 运维相关】-- HTTP 压测/负载发生器之新秀 oha
运维·网络协议·http
DONG9991 小时前
ubuntu 22 安装轻量级桌面Xfce并使用xrdp远程桌面连接
linux·运维·ubuntu
呆萌小新@渊洁1 小时前
linux升级系统,重启出现Minimal BASH-like line editingis supported
linux·服务器
東雪蓮☆2 小时前
从零开始掌握 Web 与 Nginx:入门详解
运维·服务器·前端·nginx
Miraitowa_cheems3 小时前
LeetCode算法日记 - Day 38: 二叉树的锯齿形层序遍历、二叉树最大宽度
java·linux·运维·算法·leetcode·链表·职场和发展
F_D_Z3 小时前
【PyTorch】单对象分割
人工智能·pytorch·python·深度学习·机器学习
青铜发条3 小时前
【python】python进阶——网络编程
运维·服务器·网络
浊酒南街3 小时前
Pytorch基础入门4
人工智能·pytorch·python