Ubuntu在Anaconda中安装TensorFlow GPU,Keras,Pytorch

安装TensorFlow GPU

pip install --ignore-installed --upgrade tensorflow-gpu

安装测试:

$ source activate tf    #激活tf环境
import tensorflowas tf
hello= tf.constant(\'Hello, TensorFlow!\')
sess= tf.Session()
print(sess.run(hello))
#    未报错即安装正确

安装Keras

#    直接在虚拟环境中安装
pip install keras

安装Pytorch

#    直接在虚拟环境中安装
conda install pytorch torchvision -c pytorch

安装测试:

$ source activate tf    #激活tf环境
import torch
print(torch.cuda.is_available())
#    返回True表示安装成功