小编use*_*554的帖子

Tensorflow 首次在具有 5.0 计算能力的显卡上运行需要超过 1 分钟

我正在为 python3(pip 安装)运行 tensorflow 0.8.0,以及以下文件test.py

import tensorflow as tf                                                         

a = tf.convert_to_tensor([1], dtype=tf.int32)                               
b = tf.to_float(a)                                                              

with tf.Session():                                                              
    b.eval() 
Run Code Online (Sandbox Code Playgroud)

...运行需要一分钟以上:

$time python3 test.py 
I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcublas.so locally
I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcudnn.so locally
I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcufft.so locally
I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcurand.so locally
I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:900] successful NUMA node read from SysFS had negative value (-1), but there must …
Run Code Online (Sandbox Code Playgroud)

tensorflow

4
推荐指数
1
解决办法
1076
查看次数

标签 统计

tensorflow ×1