最近看代码,用到了TensorFlow,发现服务器里没有装,这里记录一下安装过程,方便以后查阅。
Ubuntu18.04安装TensorFlow
TensorFlow安装
nvcc -V
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
创建一个虚拟环境:
conda create -n tensorflow_env python==3.7
启动虚拟环境:
conda activate tensorflow_env
在虚拟环境中安装:
pip3 install --upgrade tensorflow
安装完成测试一下:
import tensorflow as tf
print('GPU', tf.test.is_gpu_available())
a = tf.constant(2.)
b = tf.constant(4.)
print(a * b)
输出如下信息: