本人系统装了cuda12.9和cudnn,现在因为项目环境需要cuda12.6的torch2.6环境


cuda12.6

cudnn-8.9.7

cuda12.6安装过程选择自定义

只需要前三个,否则会跟高版本的组件冲突

安装完环境变量会改变注意看一下

cmd nvcc-V看一下是否改变了

安装cudnn

把全部都复制到cuda文件夹中

ok了

conda安装https://pytorch.org/get-started/previous-versions/

pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu126

测试

import torch

# 检查是否有可用的 cuda
if torch.cuda.is_available():
    # 如果有可用的 cuda,则使用 cuda
    device = torch.device("cuda")
    print("CUDA is available")
else:
    # 如果没有可用的 cuda,则使用 cpu
    device = torch.device("cpu")
    print("CUDA is not available, using CPU instead")

完成,后续改cuda就在环境变量中改变就可以了

例如12.6改12.9

还有打开path,把12.9的上移就ok了

Logo

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

更多推荐