注意:3090只能安装cuda11.x以上的版本
1)下载cuda
2) 下载cudnn
3)解压cudnn
将cudnn文件下的lib bin include 拷贝到cuda11.3目录下
4)测试
extras/demo_suite
(1) bandwidthTest.exe
(2) deviceQuery.exe
5)安装pytorch
地址:pytorch
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113