Windows10中用Anaconda安装tensorflow2.1.0(gpu版)(新手入门级)

1.安装Anaconda,添加环境变量;

anaconda官网 https://www.anaconda.com/distribution/

2.打开Anaconda navigator, 在environments下点击Create,创建独立的tensorflow环境;

 

 3.在右上角搜索框中搜索tensorflow,选中搜索结果中的tensorflow-gpu版,点击右下角的apply;

4.在设备管理器中核对电脑的GPU型号;

 5.卸载电脑原有的显卡驱动,在英伟达网站中选择对应的GPU驱动,下载安装;

https://www.nvidia.com/download/index.aspx?lang=en-us

 

 6.在anaconda navigator - environment项目下,在创建的tensorflow环境中,搜索cud,

选中cudatoolkit 和 cudnn,点击右下角的apply;

 7.在IPython中测试;

import tensorflow as tf
tf.test.gpu_device_name()
from tensorflow.python.client import device_lib
device_lib.list_local_devices()

安装过程中只有显卡驱动是从英伟达官网下载,cuda toolkit 和 cudnn都是从anaconda中下载。

参考资料:https://cloud.tencent.com/developer/article/1165267

     https://tensorflow.google.cn/install/gpu 

               

猜你喜欢

转载自www.cnblogs.com/wbloger/p/12403993.html