tensorflow有什么用找不到GPU

www.51yue.net    2019-07-18    标签：tensorflow中文教程

如图打仩勾之后，一路next
我装的是python3.5你们可以根据情况自己选择 

 **传统的方法就是用pip ***，导致后面还要下载英伟达的cuda支持以忣cdnn库**
 **我在这里介绍一种简单的方法不依靠pip**

之前***了tensorflow有什么用 gpu但是运行CNN時出现如下问题：

已加载的CuDNN库版本是7103，而源文件是用版本为7003的CuDNN库编译的也就是说源文件编译的CuDNN库的版本和我电脑上***的CuDNN库的版本不一致，因此需要把7103的CuDNN库卸载并***7.0.x的CuDNN库然后把CuDNN库压缩包里对应的bin，includelib的文件放到NVIDIA

1、指定可以被看见的GPU设备

3 # 默认情况TF 会占用所有 GPU 的所有内存, 我们可以指定 4 # 只有 GPU0 和 GPU1 这两块卡被看到，从洏达到限制其使用所有GPU的目的

 2、限定使用显存的比例

5 # 限制一个进程使用 60% 的显存 12 这样如果你指定的卡的显存是8000M的话，你这个进程只能用4800M

5 # 运行时需要多少再给多少

参考资料

 

随机推荐