dyc99

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

分类:

技术点:

相关文章:

  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-01-08
  • 2021-07-19
  • 2021-09-15
  • 2021-12-21
  • 2022-12-23
猜你喜欢
  • 2021-07-29
  • 2021-12-03
  • 2021-12-03
  • 2021-05-15
  • 2022-12-23
  • 2021-12-03
相关资源
相似解决方案