Tesla M40 下Ubuntu anaconda pycharm pytorch安装
Tesla M40 訓(xùn)練機(jī)組裝參考:http://t.csdn.cn/7Fx13,寫的很詳細(xì)
顯卡:Tesla M40 24GB (2張)
顯卡驅(qū)動版本(推薦):470.57.02
cuda版本:11.4
安裝前需要!!!!!禁用nouveau驅(qū)動!!!!!安裝顯卡驅(qū)動,該部分操作沒有太多問題,具體可參考:https://blog.csdn.net/m0_52027873/article/details/124628522
?2. 打開英偉達(dá)官網(wǎng)?下載官方cuda?
此處我下載的是cuda 11.0.3 版本,并選擇相應(yīng)配置,復(fù)制指令至終端下載?
6. 添加環(huán)境變量
? ? ?gedit ~/.bashrc打開文件,在文件結(jié)尾添加如下語句:
export PATH=$PATH:/usr/local/cuda-11.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-11.0/lib64
7. ?保存、更新環(huán)境變量
source ~/.bashrc
8.?輸入nvcc -V查看版本號
?
?借鑒Ubuntu18.04安裝Anaconda、cuda、cudnn、pytorch 、pycharm(附:保姆級圖文流程)_一點(diǎn).點(diǎn)的博客-CSDN博客_ubuntu18.04安裝anaconda
查看cudnn版本:
dpkg -l | grep cudnn?
安裝anaconda,
下載與安裝
Anaconda官網(wǎng)下載地址
測試安裝成功?
?source /root/anaconda3/bin/activate # 這個(gè)是自己的安裝目錄,以自己實(shí)際的安裝目錄為準(zhǔn) conda init
pytorch安裝
Start Locally | PyTorch
在python腳本文件中添加如下內(nèi)容指定GPU運(yùn)行程序:“0”是在GTX1070上跑的,“1”是在Tesla M40上跑的。os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"是雙卡跑的。
import os os.environ["CUDA_VISIBLE_DEVICES"] = "0"?
總結(jié)
以上是生活随笔為你收集整理的Tesla M40 下Ubuntu anaconda pycharm pytorch安装的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: ELK日志分析平台(二)----logs
- 下一篇: Ubuntu+Tesla M40上Ope