[问题] 关于Ubuntu20.04使用Pytorch问题

楼主: plps0309 (耶律阿宝机)   2021-10-08 12:37:31
最近在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
作者: brli7848 (无理阿?)   2021-10-08 13:03:00
何不直接加path...
楼主: plps0309 (耶律阿宝机)   2021-10-08 13:36:00
请问这是什么意思呢?Path的话我有按照网络教学装完cuda后放在某个路径
作者: roccqqck (ccqq)   2021-10-08 14:09:00
cudnn没装?cuda是用apt装吗
楼主: plps0309 (耶律阿宝机)   2021-10-08 14:41:00
cudnn有装 cuda我是从官网下载runfile档执行的http://i.imgur.com/qKaJuUK.jpghttp://i.imgur.com/CvcalwZ.jpg
作者: roccqqck (ccqq)   2021-10-08 15:59:00
建议删除 别用runfile装选deb比较容易升级和删除
楼主: plps0309 (耶律阿宝机)   2021-10-08 16:07:00
好的感谢回答 还是找不到问题我来试试卸载在重新安装好了
作者: roccqqck (ccqq)   2021-10-08 20:29:00
作者: creepy (左招财 右纳福)   2021-10-08 21:58:00
不要把环境弄乱掉了 还是去找docker比较好pytorch在docker hub就有官方的image
作者: brli7848 (无理阿?)   2021-10-09 03:54:00
用哪篇教学也没讲QQ path是指 env| grep PATH
楼主: plps0309 (耶律阿宝机)   2021-10-10 15:42:00
感谢各位的回答 后来全删掉后先装CUDA与CUdnn 再安装pytorch就好了 不知道是顺序还是其中可能有安装错误 我参考的教学网站如下:https://blog.csdn.net/xjgao75/article/details/105869699
作者: roccqqck (ccqq)   2021-10-16 19:22:00

Links booklink

Contact Us: admin [ a t ] ucptt.com