Question

我的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的另一种/更好的方法是什么？

使用STARTUP SCRIPT时，GCP实例上的导入张量流失败

0 个答案: