最近从theano 转到tensorflow上来,前两天一直在配环境,希望这篇博客可以帮助大家配置tensorflow-gpu

我的那台服务器忘记安装nvidia驱动直接就配好了tensorflow-cpu,首先将tensorflow卸载。

1 安装nvidia驱动

2 安装cuda-8.0,这里我曾经尝试着安装cuda9.1.发现conda install tensorflow-gpu 这个命令安装的tensorflow不能使用cuda9.1.所以只能改用cuda-8.0

3安装cudnn,这个可以参照官网的安装教程

配置tensorflow-gpu

4 配置环境变量

配置tensorflow-gpu

5 在控制台输入 conda install tensorflow-gpu 即可

注意:我这里是在 source activate (conda 环境名) 之后,在环境里安装的tensorflow-gpu

之后便可以运行了。如还有问题可以参照官网: https://www.tensorflow.org/install/install_linux

相关文章:

  • 2021-10-29
  • 2021-06-05
  • 2021-06-14
  • 2021-09-03
  • 2021-11-21
  • 2022-12-23
  • 2022-12-23
  • 2021-12-07
猜你喜欢
  • 2021-04-17
  • 2021-07-18
  • 2021-11-15
  • 2021-12-15
  • 2021-10-27
  • 2021-03-31
  • 2022-01-21
相关资源
相似解决方案