Ubuntu下安装PyTorch杂记
2022/4/24 7:13:21
本文主要是介绍Ubuntu下安装PyTorch杂记,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
最近几天我一直常用的Kubuntu(KDE yes!)更新至22.04后居然出现无法更改软件源的bug,去Kubuntu论坛一看有同样问题的人还不在少数,但却没有好的解决办法,故而只有备份数据装回Ubuntu。
由于学习需要,我需要再装一遍Pytorch。Pytorch在linux上安装推荐使用conda,使用conda安装会打包cudatoolkit等一系列必要的软件包,不过有一点值得注意,一定要换源,清华上交的都可以,而且要删除config文件里的--default,不然还是会使用默认源,我个人的计算机会在进度50多的时候显示网络问题终止安装。不过安装完成后仍有问题,第一是:
nvcc -V
显示没有安装nvidia-tool-kit,第二是因为ubuntu22.04是前天才发布的版本,nvidia驱动没有自带。总而言之
torch.cuda.is_available()
返回的是False。
对于第二个问题,很容易解决,去英伟达官网搜索自己显卡对应的驱动安装即可,而且后来我对比了一下,在终端中输入:
ubuntu-drivers devices
是可以找到官网提供的显卡驱动,我选择的是版本号510的最新稳定驱动。使用 apt-get install指令安装即可
然后使用:
nvidia-smi
可以看到自己显卡的CUDA版本,我的是11.6,虽然我在conda里确实看到了cuda等一系列包已经正确配置。没办法,我只得再去官网重新下载cudatoolkit对应的版本,照着上面的指示安装即可,安装完成后
torch.cuda.is_available()
返回值为True,这个过程大概耗了我一个下午。
这篇关于Ubuntu下安装PyTorch杂记的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-05-15PingCAP 黄东旭参与 CCF 秀湖会议,共探开源教育未来
- 2024-05-13PingCAP 戴涛:构建面向未来的金融核心系统
- 2024-05-09flutter3.x_macos桌面os实战
- 2024-05-09Rust中的并发性:Sync 和 Send Traits
- 2024-05-08使用Ollama和OpenWebUI在CPU上玩转Meta Llama3-8B
- 2024-05-08完工标准(DoD)与验收条件(AC)究竟有什么不同?
- 2024-05-084万 star 的 NocoDB 在 sealos 上一键起,轻松把数据库编程智能表格
- 2024-05-08Mac 版Stable Diffusion WebUI的安装
- 2024-05-08解锁CodeGeeX智能问答中3项独有的隐藏技能
- 2024-05-08RAG算法优化+新增代码仓库支持,CodeGeeX的@repo功能效果提升