当前位置: 首页>编程日记>正文

win10安装pytorch + cuda

1:下载cuda工具cuda-toolkit

地址:https://developer.nvidia.com/cuda-toolkit-archive

win10安装pytorch + cuda,第1张

win10安装pytorch + cuda,第2张

2:一路向下:安装

3:配置环境变量

path 环境变量中添加:

win10安装pytorch + cuda,第3张

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

nvcc -V

set cuda

5:安装cuDNN

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

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

6:安装对应pytorch版本:

Start Locally | PyTorch

win10安装pytorch + cuda,第4张选择对应的版本,安装:

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

6:验证cuda是否成功:

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

相关文章: