
GPU
湾区人工智能
会多国语言的海归
展开
-
禁用GPU版本TensorFlow,切换到CPU版本TensorFlow。
#禁用gpu版本TensorFlow,因为CUDA号码从0开始,这里直接让CUDA使用-1的GPU,自然就无法使用gpu了。 代码前面加入: import os os.environ["CUDA_VISIBLE_DEVICES"]="-1" import tensorflow as tf Environment Variable Syntax Results CUDA_VIS...原创 2018-09-11 11:17:51 · 7539 阅读 · 0 评论 -
使用指定的GPU
#使用第二个GPU, import os os.environ["CUDA_VISIBLE_DEVICES"]="1" import tensorflow as tf 使用第一个和第二个GPU import os os.environ["CUDA_VISIBLE_DEVICES"]="0,1" import tensorflow as tf...原创 2018-09-11 11:22:29 · 1299 阅读 · 0 评论 -
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 Documentation; cuDNN v6.0(Download cuDNN v6.0 (April ...原创 2018-09-25 18:29:12 · 1629 阅读 · 0 评论