win10 安装tensorflow-gpu 2.1.0版本

win10 安装tensorflow-gpu 2.1.0版本

1.安装 cuda10.1 for win10 默认安装路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
2.将cudart64_101.zip里面的文件放入C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin 下替换
3.安装cudnn:直接将cudnn-10.1-windows10-x64-v7.6.5.32 里的文件替换C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1目录下对应位置
4.pip install tensorflow-gpu=2.1.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

以上所有文件见百度网盘链接:https://pan.baidu.com/s/1pveJEr59X3AIRObkVloODg
提取码:iooo

你可能感兴趣的:(tensorflow)