1. 配置
GTX1050 系列
windows 10
VS2017
python3.7
pytorch1.2
2.准备
2.1 下载CUDA10.1,这里我选择update2
https://developer.nvidia.com/cuda-toolkit-archive
2.2 下载对应的cudnn,选择7.6.5 for 10.1,因为paddle2.1所要求
https://developer.nvidia.com/rdp/cudnn-archive
2.3 解压cudnn,将相应的inclue, lib和bin下的文件拷贝至cuda的安装目录
2.4 配置环境变量。
现在电脑装的程序越来越多,有时导致新装的程序不能将运行路径添加到系统变量path中去,导致命令行:nvcc --version无法识别,解决方法如下:
然后将cuDNN_PATH添加至path系统变量中去后重启电脑,添加后如下:
2.5 安装paddle gpu
python -m pip install paddlepaddle-gpu==2.1.2.post101 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html
一定要注意后面的后缀:2.1.2.post101,表示cuda10.1版本