이전 GPU 버전이 설치된 Tensorflow

Question

GPU가있는 공유 컴퓨터에서 작업하고 있습니다. 그것은 tensorflow-gpu==0.12.0rc1 작업 tensorflow의 한 버전을하는 데 사용하지만, 최근에는 가장 최근의 비 GPU 버전이 아니라

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 장치를 사용로 돌아갑니다.

파이썬 스크립트에서 최신 tensorflow 버전을 "숨기거나"강제로 gpu 버전을 사용하는 방법이 있습니까?

sudo 권한이없고 관리자 채널을 통과하는 데 시간이 많이 걸립니다.

Answer 1

가정에서 가상 환경을 설정하여 숨길 수 있습니다 (sudo는 필요 없음). 파이썬 버전을 지정하지 않으므로 이미 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 래퍼도 있습니다.