更新显卡驱动可以安装高版本的cuda

可以通过打开cmd输入nvidia-smi来查看
在这里插入图片描述
所以说不高于12.3的版本都可以安装

cuda版本匹配pytorch版本

打开网址：https://pytorch.org/
在这里插入图片描述
验证完后，就可以吭哧吭哧的去下载安装了
搞完后还是发型代码跑出来GPU = 0
不要慌

关键一步

在这里插入图片描述
在这里插入图片描述
把这个复制进去安装一下

运行

在这里插入图片描述

根本原因

pycharm安装的torch只支持cup计算，所以需要重新装过
在这里插入图片描述

# 深度学习
Logo
Triton中文社区

欢迎来到由智源人工智能研究院发起的Triton中文社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。

更多推荐

Stable Diffusion 针对50系显卡CUDA 12.9 的cuda error解决办法

笔者新买了RTX 5080，自然是要整AI 绘画的，但新的CUDA版本一直报错，终于在今天解决，于是写一篇文章来分享一下。

avatar Triton中文社区

jetson agx orin 刷机、cuda、pytorch配置指南【亲测有效】

2.刷机时需准备ubuntu系统，可以是虚拟机，注意安装SDK Manager刷机时，JetPack版本要选对，JetPack6.0的对应ubuntu22，cuda12版本，对应pytorch支持2.0以上的，如果你要使用pytorch2.0以下版本（pytorch1.12），则需要安装JetPack5.0，对应ubuntu20，cuda11版本，更推荐这个版本。刷机完成后，需要进行cuda环境配

avatar Triton中文社区
cover

编译支持cuda硬件加速的ffmpeg

avatar Triton中文社区