1盅蝗、環(huán)境
顯卡：Gtx 1080Ti
系統(tǒng)：Ubuntu16.04
并行：cuda11.1和對應(yīng)的cudnn8.1
軟件：Tensorflow2.4 和 PyTorch1.7
驅(qū)動(dòng)：460.39
cuda鏈接：https://pan.baidu.com/s/1_01EZN_UjQHFyr72ZeFhGA
驅(qū)動(dòng)鏈接：https://pan.baidu.com/s/1fcPakuEOeRPNaSzL1XTEKw
cudnn鏈接：https://pan.baidu.com/s/1JZcH7KDtRCuPZMx9Xk4K3g
提取碼都是：pand
2、驅(qū)動(dòng)安裝
建議先安裝驅(qū)動(dòng)，然后再安裝cuda，雖然安裝cuda的時(shí)候會(huì)帶有一個(gè)驅(qū)動(dòng)程序沧踏，但是總是會(huì)遇到錯(cuò)誤乖阵。具體方法如下：
chmod +x NVIDIA-Linux-x86_64-460.39.run
sudo ./NVIDIA-Linux-x86_64-460.39.run
然后按照下圖指示選項(xiàng)進(jìn)行選擇即可栖疑。
安裝完成，使用nvidia-smi命令驗(yàn)證結(jié)果：
3、cuda安裝
訪問https://developer.nvidia.com/zh-cn/cuda-downloads夜郁，按照下圖進(jìn)行選擇：
chmod +x cuda_11.1.0_455.23.05_linux.run
sudo sh cuda_11.1.0_455.23.05_linux.run
開始安裝什燕，如果你沒有安裝驅(qū)動(dòng)，直接安裝cuda可能會(huì)聚到下面的問題：
查看對應(yīng)的log文件發(fā)現(xiàn)安裝驅(qū)動(dòng)失敗竞端，解決方法就是按照第2步先安裝驅(qū)動(dòng)屎即，然后再安裝cuda。
如果之前安裝過cuda事富，就會(huì)遇到以下界面技俐，選擇Upgrade all。
得到結(jié)果赵颅，如下圖所示虽另，幾乎成功了。
安裝完成以后饺谬，還必須進(jìn)行以下設(shè)置：
編輯~/.bashrc捂刺，添加：
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
編輯/etc/profile，添加：
export PATH=/usr/local/cuda/bin:$PATH
創(chuàng)建鏈接文件募寨，打開sudo gedit /etc/ld.so.conf.d/cuda.conf族展，在文件中添加：
/usr/local/cuda/lib64
最后執(zhí)行 sudo ldconfig，使上述設(shè)置立即生效拔鹰。
4仪缸、cudnn安裝
在NIVIDA開發(fā)者官網(wǎng)上，找到cudnn的下載頁面: https://developer.nvidia.com/rdp/cudnn-download 列肢，選擇合適的cudnn恰画，然后在安裝完成cuda以后，執(zhí)行以下命令瓷马，就可以完成cudnn的安裝了拴还。
PS: cudnn下載必須要登錄，比較麻煩欧聘，如果和筆者配置差不多的機(jī)器片林，可以用筆者傳到百度云的鏈接下載。
cp cudnn-11.2-linux-x64-v8.1.0.77.solitairetheme8 cudnn-11.2-linux-x64-v8.1.0.77.tgz
sudo tar -xzf cudnn-11.2-linux-x64-v8.1.0.77.tgz -C /usr/local
5怀骤、Tensorflow2.4安裝
安裝最新版的非常簡單费封，直接通過pip安裝即可。
pip install tensorflow-gpu -U
如果沒有配置好驅(qū)動(dòng)蒋伦、cuda弓摘、cudnn的情況，運(yùn)行tensorflow會(huì)遇到以下問題痕届。
安裝成功的情況下就如下圖所示衣盾。
6寺旺、PyTorch 1.7 安裝
按照官網(wǎng)提示的命令進(jìn)行安裝，直接通過pip安裝即可势决，注意選擇好對應(yīng)的cuda版本阻塑。
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/torch_stable.html
