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