湾区人工智能

会四国语言的海归,公众号:《湾区人工智能》

Windows 10安装TensorFlow-gpu1.4 及CUDA8.0,cuDNN6.0,搞定了,包含安装方法和下载路径

Windows 10安装TensorFlow-gpu1.4 及CUDA8.0,cuDNN6.0,搞定了 先说下我的最终版本: Python 3.5.3; tensorflow-gpu==1.4.0 ; CUDA Toolkit 8.0 GA1 (Sept 2016), Online Doc...

2018-09-25 18:29:12

阅读数:85

评论数:0

使用指定的GPU

  #使用第二个GPU, import os os.environ["CUDA_VISIBLE_DEVICES"]="1"     import tensorflow as tf   使用第一个和第二个GPU impo...

2018-09-11 11:22:29

阅读数:87

评论数:0

禁用GPU版本TensorFlow,切换到CPU版本TensorFlow。

#禁用gpu版本TensorFlow,因为CUDA号码从0开始,这里直接让CUDA使用-1的GPU,自然就无法使用gpu了。 代码前面加入: import os os.environ["CUDA_VISIBLE_DEVICES"]="-1...

2018-09-11 11:17:51

阅读数:271

评论数:0

提示
确定要删除当前文章?
取消 删除
关闭
关闭