一、Anaconda安裝
可以參考筆者的這篇博客:Anaconda安裝詳細教程
二、準備工作
1、單擊啟動Anaconda Prompt創(chuàng)建新虛擬環(huán)境
2、在Anaconda Prompt依次執(zhí)行以下命令conda create -n pytorch python=3.6
,創(chuàng)建名字為tensorflow的虛擬環(huán)境,再通過命令conda info --envs
可以查到已存在的虛擬環(huán)境(tensorflow存在其中)
3、再使用命令conda activate tensorflow
激活剛剛建立好的虛擬環(huán)境tensorflow,可以看見小括號中由base環(huán)境切換到tensorflow環(huán)境
三、TensorFlow安裝
1、TensorFlow版本選擇
TensorFlow版本分為CPU版本和GPU版本,CPU版本一般電腦都可以安裝,但是GPU版本只有電腦的顯卡支持cuda才可以安裝,這兩個版本的區(qū)別主要在于運行速度,GPU版本的TensorFlow運行速度更快,所以若電腦顯卡支持cuda,推薦安裝GPU版本。本文以安裝GPU版本的TensorFlow進行說明。
2、TensorFlow的GPU版本對應Python版本
所有博客都建議去Tensorflow文檔查看,其網(wǎng)址為https://tensorflow.google.cn/install/source_windows
,但是筆者確沒有找到。根據(jù)筆者安裝的經(jīng)驗,TensorFlow 1.x的GPU版本對應的Python版本在3.7以下,TensorFlow 2.0和TensorFlow 2.1的GPU版本對應的Python版本在3.7以下,TensorFlow 2.2的GPU版本對應的Python版本在3.8以上。
筆者建議在創(chuàng)建虛擬環(huán)境時即指定Python版本為3.6版本既可向下兼容TensorFlow 1.x的GPU版本,又可以兼容TensorFlow 2.x的GPU版本。此處指出筆者安裝TensorFlow 的GPU版本時出現(xiàn)的一個問題,因為筆者的base環(huán)境的Python版本為3.8版本,在創(chuàng)建虛擬環(huán)境時默認為Python版本為3.8。隨后直接安裝了最新的TensorFlow 2.3 的GPU版本,在進行測試時輸入語句
import tensorflow as tf
直接報錯:module 'numpy' has no attribute 'object'
,具體原因不可知。
3、TensorFlow的GPU版本安裝
(1)在剛才最小化的Anaconda prompt窗口中輸入命令conda install -c aaronzs tensorflow-gpu==1.15.0(注意這是TensorFlow 1.x的最后一個版本)
(2)待下載完畢后,輸入python
,會顯示python版本為3.6。之后進行TensorFlow的測試,輸入以下代碼:
import tensorflow as tf
import os
a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)
print(tf.__version__)
print('GPU:', tf.config.list_physical_devices('GPU'))
print(tf.test.is_gpu_available())
輸出結果為:
Tensor("add:0", shape=(), dtype=float32)
1.15.0
# 筆者在測試第三個print時有問題,大家可以自行測試
AttributeError: module 'tensorflow._api.v1.config' has no attribute 'list_physical_devices'
True
無需安裝cuda和cuDNN!!
參考博客:文章來源:http://www.zghlxwxcb.cn/news/detail-816652.html
(1)TensorFlow-GPU極簡———不用安裝cuda、cuDNN?。?!
(2)使用anaconda安裝tensorflow (windows10環(huán)境)
(3)Anaconda安裝使用及tensorflow配置文章來源地址http://www.zghlxwxcb.cn/news/detail-816652.html
到了這里,關于使用Anaconda安裝TensorFlow詳細教程的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關文章,希望大家以后多多支持TOY模板網(wǎng)!