pytorch离线下载网
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
torchvision离线下载网
https://download.pytorch.org/whl/torchvision/
torchaudio离线下载网
https://download.pytorch.org/whl/torchaudio/


教程网址 CUDA安装教程(超详细)
https://blog.csdn.net/m0_45447650/article/details/123704930

torch下载网址
https://download.pytorch.org/whl/torch_stable.html

NVIDIA RTX A2000 12G显卡驱动
https://www.nvidia.cn/Download/index.aspx?lang=cn

cuda版本号
cuda_11.4.0_471.11_win10
下载网址
https://developer.nvidia.com/cuda-11-4-0-download-archive?target_os=Windows&target_arch=x86_64&target_version=Server+2019&target_type=exe_local

点击设置-->搜索高级系统设置-->查看环境变量

win+r,cmd
输入nvcc --version 即可查看版本号;
set cuda
可以查看 CUDA 设置的环境变量。

cudnn版本号
cudnn v8.9.3(July 11th,2023),for cuda 11.x
https://developer.nvidia.com/rdp/cudnn-download
把三个文件夹拷贝到cuda的安装目录下
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0
系统变量---系统环境变量中的 path 添加如下路径(根据自己的路径进行修改)
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\libnvvp

win+r,输入cmd
cd C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.4/extras/demo_suite
$bandwidthTest.exe
$deviceQuery.exe

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐