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表示安装成功