在安装前首先确认CUDA的版本,并将其与tensorflow所需的libcudart库版本匹配。下面是一些可能有用的代码示例:
!nvcc --version
import tensorflow as tf print(tf.sysconfig.get_lib())
sudo apt-get install libcudart10.1
sudo ln -s /usr/local/cuda/lib64/libcudart.so.10.1 /usr/lib/x86_64-linux-gnu/libcudart.so.10.1
import tensorflow as tf hello = tf.constant('Hello, TensorFlow!') sess = tf.Session() print(sess.run(hello))