PythonにおけるCUDAを利用したGPU制御について
PythonにてTensorflow-gpuを利用したいと思い、CUDA ToolkitやcuDNNのインストールを行い、GPUを用いてプログラムを実行できていることを確認しました。
私のPCには、GPUが2台搭載されており、使用するGPUを指定したいと思っています。
プログラム内で具体的に
with tf.device(~):
と記述することで、使用するGPUを指定できることは確認しました。
それとは別に,プログラム実行時に、コマンドプロンプト上で
CUDA_VISIBLE_DEVICES=1 python ~.py
とすることで使用するGPUを制限できるというのを見かけました。
幾つかのWebサイトを見たのですが、上記のコマンドを実行すると
'CUDA_VISIBLE_DEVICES' は、内部コマンドまたは外部コマンド、
操作可能なプログラムまたはバッチ ファイルとして認識されていません。
と言われます。どうすればこのコマンドが使用できるようになりますか?
PCのシステム面への知識不足だとは思うのですが、よろしくお願いします...
実行環境 : Windows 8.1