TensorFlow学习(11)安装GPU版本的Tensorflow

安装CUDA

1. 准备好NVIDIA的显卡,下载安装CUDA:https://developer.nvidia.com/cuda-downloads

2.安装好之后把CUDA安装目录下的binlib\x64添加到环境变量Path中

TensorFlow学习(11)安装GPU版本的Tensorflow_第1张图片

安装cuDNN

1. cuDNN下载:https://developer.nvidia.com/rdp/cudnn-download 

2. 解压压缩包,把压缩包中bin,include,lib\x64中的文件分别拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1

3. 把C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\lib64\cupti64_101.dll拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin

然后卸载原来版本的tensorflow

pip uninstall tensorflow

安装gpu版本的tensorflow

pip install tensorflow-gpu

你可能感兴趣的:(深度学习)