相关概念

CUDA 有两种API,分别是驱动API和 运行时 API 。

nvidia-smi:如果能正常使用,则说明Driver API 已经安装,nvidia-smi显示的是支持的最高的cuda版本,是向下兼容的
nvcc -V :是指实际安装的cuda toolkit 必须小于等于driver API支持的最高版本。

cuda卸载

非常感谢这篇博客,没有它我一天都卸载不了

cuda安装

cuda安装教程1
cuda安装教程2
cuda 官网

安装pytorch

https://pytorch.org/get-started/previous-versions

我遇到的问题

我的服务器镜像已经安装了GPU驱动和cuda10.2(cuda toolkit) ,但是RTX 3090 的算力是sm_86,cuda10.x是不支持这个算力的,因此需要安装cuda11.x才可以。

RTX3090算力对应关系

Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐