linux manjaro 配置 pytorch gpu 環境


manjaro目前中國資料偏少,踩了很多坑。

安裝gpu版本就這么幾個步驟 1 安裝英偉達的驅動cuda  2 安裝 cudnn   3 安裝支持gpu的pytorch 或者其他的運算框架

 

manjaro中文資料很少,,,一步一步先裝cuda 再裝cudnn  好麻煩,各種錯誤

 

實際上就這一句就解決了:

1 裝 cuda 和 cudnn

sudo pacman -S python-pytorch-cuda

就這一句就自動吧 cuda cudnn 都裝好了,

這句話執行中間會失敗很多次,,多幾次重新執行,它把所有包下載好了 就能成功了。


然后使用命令查看一下自己的cuda 是什么版本
optirun nvidia-smi

 

 

2 安裝支持gpu的pytorch

去官方網站 https://pytorch.org/

找對應自己的pyorch安裝

 

我的cuda 是10.1  但是我發現網站上pytorch 支持cuda的只到10, 所以我就安裝了cuda10的pytorch。。。。后來也成功了

 

3  驗證一下

用  optirun  python 開python交互,,

import torch

torch.cuda.is_available()

 

就成功了。

 

據說manjaro系統不會自動切換獨立顯卡,,所以運行時候要加上optirun 才能調用獨立顯卡

 

 

 
       


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM