前言
從tensorflow1.X用到現(xiàn)在了现斋，pytorch也是去年接觸的吟孙，這兩個(gè)框架都屬于更新比較快的讨阻，因此難免更新自己的版本芥永。最頭疼的莫過(guò)于CUDA，cudnn這些東西的版本匹配钝吮。以前看了不少教程埋涧，讓我們安裝cuda，cudnn奇瘦，配置環(huán)境變量棘催。其實(shí)，這兩個(gè)框架的GPU版本配置早就很簡(jiǎn)單很簡(jiǎn)單了耳标！根本不用下載CUDA醇坝，cudnn這些！
準(zhǔn)備工作
要準(zhǔn)備的就兩點(diǎn)：
1.Anaconda/Miniconda的安裝
2.GPU驅(qū)動(dòng)版本的確認(rèn)
首先是Anaconda次坡，你可以下載官方的呼猪，也可以通過(guò)清華鏡像下載，個(gè)人推薦miniconda砸琅，啟動(dòng)快宋距，功能都有，不用修改環(huán)境變量症脂，這些統(tǒng)統(tǒng)都是多余的谚赎！接下來(lái)建議采用conda安裝pytorch這些，因此可以換conda清華源诱篷，具體怎么做點(diǎn)進(jìn)去有教程壶唤，如果換源后出現(xiàn)問(wèn)題可以去掉“-default”，把https改為http棕所。
其次是驅(qū)動(dòng)闸盔，首先你要有N卡，其次你要確定N卡是否支持橙凳，可以通過(guò)谷歌搜索“你的GPU型號(hào)+SPECIFICATION”確定是否支持cuda蕾殴。
pytorch安裝
打開(kāi)pytorch官網(wǎng)笑撞，調(diào)整好自己所需的內(nèi)容：
注意：
1.如果你用清華源需要去掉 “-c pytorch”；
2.cuda 10.2以上版本需要vs編譯環(huán)境的支持钓觉，最簡(jiǎn)單的方法是安裝vs茴肥，也可以安裝vs developtoolkit
以上，裝完了荡灾，不用配置cuda瓤狐，都是多余的，因?yàn)槟阍诎惭b時(shí)conda會(huì)自動(dòng)安裝批幌。測(cè)試是否安裝成功：
activate “環(huán)境名”
python
import torch
torch.cuda.is_available()
正常的話就會(huì)輸出True
tensorflow安裝
打開(kāi)官網(wǎng)础锐，官網(wǎng)沒(méi)conda指令，你可以：
conda install tensorflow-gpu
在安裝cudatoolkit和cudnn荧缘，注意要與你安裝的tf對(duì)應(yīng)：
conda install cudatoolkit=10.1
conda install cudnn=7.6.5
測(cè)試皆警，結(jié)束。
