Google Colaboratory上でPythonで作成したGANのコードを動作させようとしています。
コードは、以下の環境で動作させています。
・Python:3.6
・Tensorflow:tensorflow-gpu 1.14.0
・Keras:2.1.3
上記環境で動作はするのですがGPUが認識できず、動作が遅いです。
調べたところ、Tensorflow、CUDA、cudnnのバージョンがTensorflow GPU, CUDA, CuDNNのバージョン早見表
の通りの組み合わせになっていないとGPUで動かないことが分かりました。
現在のGoogle Colaboratoryでインストールされているのは
・CUDA:10.1.243
・cudnn:7.6.5
なので、これをtensorflow-gpu 1.14.0でGPUが動くよう、
・CUDA:10.0
・cudnn:7.4
にダウングレードしたいです。
Google Colaboratoryに入っているCUDAとcudnnはダウングレードできるものなのでしょうか?
もしできるのであれば、その方法をお教えして頂きますと助かります。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2020/05/09 12:55