仅用以记录个人环境配置流程。

一、安装CUDA Driver

1.1 查看CUDA Driver版本

1.2 驱动更新下载地址：NVIDIA Studio 驱动程序 | 512.59 | Windows 11

CUDA Driver即显卡驱动版本，高于CUDA Toolkit 指定的版本，也是能正常使用的。

二、安装CUDA Toolkit

2.1 查看cuda toolkit与驱动版本对应关系

查看地址：Release Notes :: CUDA Toolkit Documentation (nvidia.com)

下载地址：CUDA Toolkit Archive | NVIDIA Developer

下载：

 安装路径：C:\Users\lion\AppData\Local\Temp\CUDA

三、安装cudnn

下载地址：cuDNN Archive | NVIDIA Developer

下载完成后，将cudnn文件夹下的文件复制到cuda安装目录

四、配置环境变量

 

 

 五、安装tensorflow-gpu 与keras

5.1查看版本对应关系：Build from source on Windows  |  TensorFlow (google.cn)

5.2安装keras

安装tensorflow==2.8.0 版本太新出现不匹配的问题，退到2.6.0

tensorflow出现报错：Could not load library cudnn_cnn_infer64_8.dll. Error code 126

解决方案：Installation Guide :: NVIDIA Deep Learning cuDNN Documentation​​​​​​

安装zlib，解压后将zlibwapi.dll添加到环境变量。

六、安装pytorch

一行搞定：pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

cu110指的是11.0版本的cuda 

轮子下载地址：https://download.pytorch.org/whl/torch_stable.html

本机安装 ： cuda 11.6 - cudnn11.5   tensorflow==2.6.0 /pythoch==1.7.1

# 深度学习 # 人工智能
Logo
FlagOS智算系统软件栈

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

更多推荐

PyTorch CUDA调试第一步：5分钟学会使用torch_use_cuda_dsa

是PyTorch提供的一个调试工具，它允许你在CUDA设备端（GPU）执行断言检查。简单来说，就是在GPU上运行的代码中加入断言语句，当条件不满足时会触发错误，帮助你快速发现代码中的问题。这对于调试CUDA内核中的错误特别有用，因为设备端的错误通常比主机端更难调试。是一个非常实用的调试工具，尤其适合CUDA内核的调试。通过设备端断言，你可以快速发现代码中的逻辑错误，提高调试效率。希望这篇笔记能帮助

avatar FlagOS智算系统软件栈

如何用AI优化PyTorch CUDA调试：torch_use_cuda_dsa详解

例如，在矩阵乘法中，可以断言矩阵的维度匹配，或者在计算过程中检查中间值是否在合理范围内。通过AI辅助工具，如Kimi-K2模型，我们可以快速生成带有详细注释的代码示例，解释每个参数的作用和调试技巧。AI不仅能帮助我们理解复杂的CUDA调试技术，还能提供自动补全和错误诊断功能，显著提高开发效率。为了更好地理解断言的作用，我们可以故意在代码中引入一些可能触发断言的条件。对于需要进行CUDA调试的开发者

avatar FlagOS智算系统软件栈

解决bitsandbytes安装难题：libcudart.so找不到的终极方案

🚀 **bitsandbytes** 是一个革命性的PyTorch库，通过8位量化技术让大型语言模型变得触手可及。这个强大的工具能够将模型推理和训练的内存消耗降低到原来的几分之一，但安装时经常遇到的"libcudart.so not found"错误让许多开发者头疼不已。今天，我将为你提供一套完整的解决方案，彻底告别这个困扰！## 🔍 为什么会出现libcudart.so找不到的错误？

avatar FlagOS智算系统软件栈