GPU深度学习

一 下载与对应的cuda和cudnn

需要下载的cuda版本小于12.4（下载了cuda11.8，并下载对应的cudnn），若cuda版本小于11.8需要再NVIDIA官网下载驱动程序并安装

二 安装顺序

1.安装VS（VisualStudioSetup）本体，不用添加任何负载
2.安装CUDA
3.将cuDNN中的bin、include、lib3个文件夹复制到CUDA目录下，替换操作

三 安装是否成功检验

import torch
print(torch.cuda.is_available())   #输出为true，则安装成功

参考链接：https://blog.csdn.net/qq_49641239/article/details/118175701
我这里安装最低的版本

pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
# 使用清华源下载其他所有的库
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple