["CUDA_VISIBLE_DEVICES"]="0" 沒有使用到GPU? - Cupoy

2019/07/18 下午 01:29

機器學習共學討論版

Edwin

觀看數：271

回答數：3

收藏數：0

ml100-2

ml100-2-d78

在train model的時候

os.environ["CUDA_VISIBLE_DEVICES"] = "0"和

os.environ["CUDA_VISIBLE_DEVICES"] = ""

所花費的時間一樣，不知道會是甚麼原因

CPU:AMD Ryan 5 1600

GPU: GTX 1080 

RAM: 16GB

回答列表

• 2019/07/19 上午 01:19
  Seanyu.TW
  贊同數：0
  不贊同數：0
  留言數：0

  Hi,

  有幾個項目可能麻煩你確認一下

  1. 是否安裝 cuDNN

  2. tensorflow-gpu 的版本號

  3. 訓練時間一樣是在哪個資料集上，以及將 CUDA_VISIBLE_DEVICES 設為 "" 與 "0" 時的速度分別為何呢

• 2019/07/19 上午 10:37
  Jeffrey
  贊同數：0
  不贊同數：0
  留言數：2

• 2019/07/19 上午 10:40
  Edwin
  贊同數：0
  不贊同數：0
  留言數：1

  1.有安裝，x64 應該就是cudnn的lib folder 

  

  2. tensorflow-gpu version = 1.13.1

  

  3.

  訓練集與這天(day78)作業相同

  設為""  = 237 sec 

  設為 "0"  = 240 sec