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