windows10+anaconda3+python3+tensorflow-gpu+cuda+cudnn 配置tfgpu平台

xiaoxiao2021-02-28  35

基本是按照这一篇博客来的,这篇有一定辅助,总体比较顺利,因为之前装过cpu版的

其中安装cuda 的细节按这篇

有一些踩坑点:

1.python3.5 一般对应tf1.3.0 对应cuda8.0,对应cudnn 6.0 (这个那篇博客里没说,cudnn现在直接登上去是7.0版的了,7.0版是不行的,会导致 报错  ImportError: No module named '_pywrap_tensorflow_internal',网上有人说是vs2015没装,其实不是的,我就没装,是cudnn版本问题

2.windows下查看自己电脑配置的nvidia显卡配置信息的命令 记录如下

C:\Program Files\NVIDIA Corporation\NVSMI\nvidia-smi.exe (自动安装路径的情况下)

XXX\nvidia-smi.exe(xxx是cuda的安装路径)  (自己选择路径的情况下)

(其实与ubuntu上的$nvidia-smi 命令是一样的,小编最近一直在ubuntu下装N卡,现在还没成功。。。)

当然,注意先cd 再运行程序

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

最新回复(0)