[問題] 關於Ubuntu20.04使用Pytorch問題

[plps0309 (耶律阿寶機)]

最近在Ubuntu下安裝pytorch遇到一些關於CUDA的問題

已經安裝過CUDA 11.1以及CUDA Toolkit 11.1

但在python下輸入print(torch.cuda.is_available())卻總是顯示False

想請問是否有哪個步驟有遺漏掉呢？
http://i.imgur.com/6qFJuFb.jpg
http://i.imgur.com/TJhsd9g.jpg
http://i.imgur.com/uiZgbPh.jpg

-----
Sent from JPTT on my Google Pixel 5.

--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 114.43.173.221 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/Linux/M.1633667853.A.EE6.html
→ brli7848: 何不直接加path...
→ plps0309: 請問這是什麼意思呢？Path的話我有按照網路教學裝完cuda
→ plps0309: 後放在某個路徑
推 roccqqck: cudnn沒裝?
→ roccqqck: cuda是用apt裝嗎
→ plps0309: cudnn有裝 cuda我是從官網下載runfile檔執行的http://i.
→ plps0309: imgur.com/qKaJuUK.jpg
→ plps0309: http://i.imgur.com/CvcalwZ.jpg
推 roccqqck: 建議刪除 別用runfile裝
→ roccqqck: 選deb比較容易升級和刪除
→ plps0309: 好的感謝回答 還是找不到問題我來試試卸載在重新安裝好
→ plps0309: 了
推 roccqqck: https://pytorch.org/get-started/locally/
推 creepy: 不要把環境弄亂掉了 還是去找docker比較好
推 creepy: pytorch在docker hub就有官方的image
→ brli7848: 用哪篇教學也沒講QQ path是指 env| grep PATH
→ plps0309: 感謝各位的回答 後來全刪掉後先裝CUDA與CUdnn 再安裝pyt
→ plps0309: orch就好了 不知道是順序還是其中可能有安裝錯誤 我參考
→ plps0309: 的教學網站如下：
→ plps0309: https://blog.csdn.net/xjgao75/article/details/105869
→ plps0309: 699
→ roccqqck: https://github.com/veritas9872/PyTorch-Universal-Doc
→ roccqqck: ker-Template