搜索安裝包並選擇合適版本:
# 搜索當前可用的安裝包版本
anaconda search -t conda tensorflow
# 選擇合適版本,查詢安裝命令
anaconda show anaconda/tensorflow-gpu
注意tensorflow的版本要與CUDA匹配,我是CUDA10,下載了tensorflow1.12
結果:
按照最後一行所示命令進行安裝
conda install --channel https://conda.anaconda.org/anaconda tensorflow-gpu
測試
$ python
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, Tensorflow')
>>> sess = tf.Session()
>>> print(sess.run(hello))
終於成功了嗚嗚嗚------------------------------------------
參考: