1

我的 GCP GPU 实例启动脚本安装cuda + cudnn + tensorflow-gpu并编码如下：

## 1. Steps to install CUDA ##
## 2. steps to setup cudnn ##

## 3. set PATH & LD_LIBRARY_PATH variables
echo 'export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}' >> ~/.bashrc 
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc 

## 4. install tensorflow-gpu using pip
## 5. call python script which will 'import tensorflow'

我看到脚本没有正确完成第 3 步.bashrc，并且文件不包含这两个条目。此外，出于类似的原因，我在导入 tensorflow 的语句中收到以下错误：

INFO startup-script:     import tensorflow as tf
INFO startup-script:   File "/usr/local/lib/python2.7/dist-packages/tensorflow/__init__.py", line 24, in <module>
.
.
INFO startup-script: ImportError: libcudnn.so.7: cannot open shared object file: No such file or directory
INFO startup-script: Failed to load the native TensorFlow runtime.

如果我在控制台窗口中手动导出这两个变量，然后运行 ​​python 脚本（步骤 5）；有用。

启动脚本自动设置这些变量的方法是什么？

或者通过 gcp 上的启动脚本使用 tensorflow-gpu 的其他/更好的方法是什么？

4

0 回答 0