GTX960M安裝tensorflow-gpu心得

GTX960M安裝tensorflow-gpu心得與體會

安裝了很多次，又卸載了很多次，終于成功了，不容易啊

安裝Anaconda

由于電腦不算太新，所以Anaconda安裝的是python3.6版本。
anaconda安裝 ，安裝教程很多，記得把自動添加到環(huán)境變量的那處打鉤

安裝cuda

cuda安裝的版本是10.1，這個可以去網(wǎng)站上找很多的

安裝失敗解決辦法，取消掉CUDA里Visual Studio Integration前的√，其他可以全選，然后安裝。之后再次安裝cuda，不過只安裝Visual Studio Integration安裝就可以了。
安裝成功使用 nvcc --version 查看，如果顯示就成功了。

安裝cudnn

cudnn選擇的是7.6版本，將里面的文件都復(fù)制到
**NVIDIA GPU Computing Toolkit ** 這個文件夾對應(yīng)得文件里面

安裝tensorflow-gpu

安裝的tensorflow-gpu版本是2.1.0，與之對應(yīng)的cuda10.1,cudnn是cuda10.1相對應(yīng)的其中的7.6版本，

第一步在命令行Anaconda Prompt 里面創(chuàng)建一個虛擬環(huán)境
conda create -n tf-gpu python=3.6

激活虛擬環(huán)境 activate tf-gpu

在虛擬環(huán)境里面安裝tensorflow-gpu pip install tensorflow-gpu==2.1.0

安裝完之后進(jìn)入python命令行驗證
import tensorflow as tf
a=tf.constant([1.0,2.0,3.0],shape=[3],name=‘a(chǎn)’)
b=tf.constant([1.0,2.0,3.0],shape=[3],name=‘b’)
c=a+b
#通過log_device_placement參數(shù)來輸出運行每一個運算的設(shè)備
sess=tf.Session(config=tf.ConfigProto(log_device_placement=True))
print (sess.run?)

沒什么問題后可以使用 deactivate退出環(huán)境

在pycharm使用tensorflow-gpu

創(chuàng)建新項目，新解虛擬環(huán)境，不過基礎(chǔ)解釋器需要使用tf-gpu，這個在anaconda安裝目錄下單envs里面找到，選擇tf-gpu里的python.exe就可以了。創(chuàng)建的所有虛擬環(huán)境都會存放在envs文件夾里面。之后點擊create就可以了。

如果希望tf-gpu解釋器里面包含的庫更多，可以進(jìn)入命令行的tf-gpu虛擬環(huán)境，使用pip install 。。。 安裝你需要的庫，以后就不要新的項目里面去添加了，所有繼承tf-gpu的虛擬環(huán)境都可以使用了。

所有東西放在百度云盤上面了。方便下載自取。tensorflow-gpu 2.1.0這個使用命令行就可下載了。
鏈接：https://pan.baidu.com/s/1NWfwdxc6P-SP9oWTlVT6Bg
提取碼：tfcu
里面包含
Anaconda3-5.2.0-Windows-x86_64
cuda_10.1.243_426.00_win10
cudnn-10.1-windows10-x64-v7.6.5.32

總結(jié)

以上是生活随笔為你收集整理的GTX960M安装tensorflow-gpu心得的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。