系统:win10 已安装:python3.10&pycharm 显卡:MX450独立显卡
一、(仅独显且安装至GPU需要,其余情形可直接跳过)安装CUDA与CUDNN
1.注意事项
CUDA、CUDNN、以及Anaconda虚拟出的环境中所需要安装的PyTorch之间是有版本上的对应关系的,由于下载CUDNN需要在NVIDIA上注册账户且NVIDIA网站似乎很不稳定,因此这里我的做法是在清华的镜像上下载CUDNN。
另外,通过Nvidia控制面板->帮助->系统信息-> 组件 处查看支持的CUDA版本,假设支持的CUDA版本是11.2,那么就可以安装版本号<=11.2的CUDA。
以下是安装链接:
CUDA(版本10.2):CUDA Toolkit 10.2 Download | NVIDIA Developer
【注意】Installer Type一栏建议选择exe(local)
CUDNN(清华镜像,注意下载的CUDNN应当对应于CUDA10.2,从文件名上即可看到):