在windows上安装tensorflow gpu

xiaoxiao2021-02-28  108

近期,我在电脑上安装了tensorflow gpu版本,遇到一些问题,在这里写一些解决方案。

1、安装python3

2、安装tensorflow gpu:pip install tensorflow-gpu

3、查看显示适配器中关于NVIDIA,https://developer.nvidia.com/cuda-gpus,查看是否支持cuda

4、下载cuda,https://developer.nvidia.com/cuda-downloads,根据我的安装,发现tensorflow只支持cuda8.0

5、下载cudnn v5.1,cudnnv6.0安装完后会报错,将cudnn中lib,include,bin中文件移到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0 ,可以根据你的路径变换。

6、测试,打开python,输入import tensorflow as tf,如无报错,则安装成功

7、如果报错 No module named '_pywrap_tensorflow_internal',则是因为缺少msvcp140.dll,需要安装visual c++ 2015 redistributable Package,https://www.microsoft.com/en-us/download/details.aspx?id=52685

转载请注明原文地址: https://www.6miu.com/read-26308.html

最新回复(0)