2017-03-16 18 views
0

私はGPUを使用して共有マシンで作業しています。それはtensorflow-gpu==0.12.0rc1作業tensorflow 1つのバージョンだけを持っていたが、最近、それはまた、最新の非GPUのバージョンを持っているだけでなく古いGPUバージョンがインストールされたTensorflow

tensorflow==1.0.1 
tensorflow-gpu==0.12.0rc1 

は私のコードはもうGPUデバイスを使用しません。私は古いバージョンのみimport tensorflow-gpu as tfをインポートまたは古いバージョンであることをtensorflow-gpuを要求しようとした:

import pkg_resources 
pkg_resources.require("tensorflow-gpu==0.12.0rc1") 
import tensorflow as tf 

それだけでCPUのデバイスを使用してに戻ります。

は私のPythonスクリプトから最新tensorflowバージョンを「隠す」またはいずれかの方法で、GPUのバージョンを使用するように強制する方法はありますか?

私はsudo特権を持っていません。管理用チャネルを経由するには時間がかかります。

答えて

1

あなたの家に仮想環境を設定することで隠すことができます(sudoは必要ありません)。あなたはあなたのpythonバージョンを指定していないので、すでにvirtualenvがインストールされていると仮定します(それは最新バージョンが付属しています)。そうでなければインストールします(もう一度、sudoは必要ありません - https://stackoverflow.com/a/5177027/524436参照)。その後

# create directory to store environments in 
mkdir -p ~/virtualenvs 
cd ~/virtualenvs 

# create virtual environment for tensorflow 
# --no-site-packages hides the global packages like the CPU TF 
virtualenv --no-site-packages tensorflow 

# activate the virtual environment 
source tensorflow/bin/activate 

であなたが開くので、もしあなたがパイソン

# install what you want - tensorflow, matplotlib, ... 
pip install matplotlib 

のために「活性化」の行をパッケージをインストールすることができ、クリーンな環境でのみ、現在のシェルセッションを変更している得ますあなたがそれをアクティブにするためにsource ~/virtualenvs/tensorflow/bin/activateを行う必要がある新しいシェル。 virtualenvには、必要な型の量を減らすためのラッパーもあります。

+0

完璧な、これは私の問題を解決しました。どうもありがとう! –

関連する問題