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ファイルに 2 つのエントリが含まれていないことがわかりました。また、同様の理由で、テンソルフローをインポートするステートメントで以下のエラーが発生します。
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.
コンソール ウィンドウで 2 つの変数を手動でエクスポートし、python スクリプトを実行すると (ステップ 5); できます。
これらの変数を起動スクリプトによって自動的に設定する方法は何ですか?
または、 gcp の起動スクリプトを介して tensorflow-gpu を使用する他の/より良い方法は何ですか?