win10安装pytorch + cuda

作者：花生_TL007 | 2024-05-05 20:58:36

踩

1：下载cuda工具cuda-toolkit

2：一路向下：安装

path 环境变量中添加：

4：验证cuda是否安装成功：

set cuda

5:安装cuDNN

登录cuDNN下载网页（必须注册账号并登录）并安装
cuDNN Archive | NVIDIA Developer

将解压后的文件夹中的bin、include、lib 三个文件夹，移动到CUDA Development 安装路径下，
与同名文件夹合并。

6：安装对应pytorch版本：

选择对应的版本，安装：

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

6：验证cuda是否成功：


import torch
print(torch.cuda.is_available())

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/花生_TL007/article/detail/540811