目录
项目场景
在新组装的台式机安装windows和linux双系统,安装pytorch的GPU版本。
windows系统装gpu版本的pytorch见另一篇文章【windows10下安装GPU版pytorch简明教程,仅需两步骤,以及解决报错:pytorch 无法导入torch加载..\caffe2_nvrtc.dll”或其依赖项之一时出错】-CSDN博客
1、问题描述
1、vmware虚拟机版本的ubuntu系统性能不好,而且知乎上说虚拟机版本的linux不能使用gpu版的pytorch,所以需要装双系统
2、如何安装
安装windows和linux双系统
3、如何在linux系统,即ubuntu系统装gpu版本的pytorch
2、安装windows和linux双系统
参考B站一个UP主的保姆级教程:Windows 和 Ubuntu 双系统的安装和卸载,作者:机器人工匠阿杰。未经作者允许,不能转载。所以这里只列出视频名字和作者了。
1、如果安装完毕重启后遇到usb问题,可能是台式机无线网卡不适用ubuntu系统
3、ubuntu系统装gpu版本的pytorch
需要安装anaconda、显卡驱动、cuda
参考文献:
1、ubuntu系统装显卡驱动
# 查看推荐版本
1、命令行输入:
sudo ubuntu-drivers devices
然后显示可选的版本
driver: nvidia-driver-525-open - distro non-free
driver: nvidia-driver-470-server - distro non-free
driver: nvidia-driver-470 - distro non-free
driver: nvidia-driver-515 - distro non-free
driver: nvidia-driver-525 - distro non-free
driver: nvidia-driver-530 - distro non-free
driver: nvidia-driver-515-server - distro non-free
driver: nvidia-driver-525-server - distro non-free
driver: nvidia-driver-515-open - distro non-free
driver: nvidia-driver-510 - distro non-free
driver: nvidia-driver-530-open - distro non-free recommended
driver: xserver-xorg-video-nouveau - distro free builtin
不要选择带open的版本可能会有新的问题
列表中会有一行最后带有“recommended”,这是系统推荐版本
2、选择一个版本安装
sudo apt install nvidia-driver-535
3、快结束的时候(可能)会让输入一个secure boot的密码(Configuring Secure Boot 界面,一定要记住这个密码),在后面会用到,尽量简单一些,后面是盲输。重启电脑,这一步很重要,不重启没有效果。(
这一部分参考下面的链接【精选】Ubuntu18.04 配置nvidia 460驱动、cuda10.1和cudnn7.6.5_☞小星☜的博客-CSDN博客
————————————————
版权声明:本文为CSDN博主「☞小星☜」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_52297947/article/details/125712327)
4、重启电脑
sudo reboot
5、重启就会出现一个蓝色的perform mok management 界面
这一部分参考下面的链接【精选】Ubuntu18.04 配置nvidia 460驱动、cuda10.1和cudnn7.6.5_☞小星☜的博客-CSDN博客
选择第二个enroll mok,然后选择continue,再选择yes,就会出现让输入密码的界面,准确输入你之前设定的secure boot密码,然后选择reboot。在输入密码时,是暗输入的,因此不必惊慌,确保密码正确即可。
在进入这个界面时,一定不要直接点击continue boot,否则会导致加载nvidia的驱动失败。如果点错后,则需要卸载并重新装一个nvidia驱动,然后再次重启电脑。
6、重启后,输入命令查看nvidia驱动是否安装成功:
nvidia-smi
2、安装cuda
先判断自己系统ubuntu是x86还是arm64架构
2.1 检查操作系统版本
我们首先需要检查当前操作系统的版本信息。为了实现这一步骤,我们可以使用以下命令:
lsb_release -a
这个命令会显示当前操作系统的版本和其他相关信息。
2.2 检查架构类型
接下来,我们需要检查操作系统的架构类型。我们可以使用以下命令来实现这一步骤:
uname -m
这个命令会输出操作系统的架构信息。
2.3 输出结果
如果操作系统版本为Ubuntu且架构类型为x86_64,则说明Ubuntu是x86架构。如果操作系统版本为Ubuntu且架构类型为aarch64,则说明Ubuntu是arm64架构。
2.3.安装步骤
具体安装步骤参考
【精选】Ubuntu18.04 配置nvidia 460驱动、cuda10.1和cudnn7.6.5_☞小星☜的博客-CSDN博客
2.3.1 决定pytorch的版本以此来选择cuda版本
Pytorch传送门:Pytorch
CUDA传送门:CUDA
2.3.2、其中可能出现的问题
1、安装cuda时出现根目录空间不足
(1)方法1:安装到其他地方
ubuntu 20.04LTS下安装cuda时提示/tmp空间不足问题的解决方案_cuda因为磁盘空间不够未安装完成咋办-CSDN博客
(2)方法2:根目录扩容
Ubuntu20.04根目录扩容(双系统)-CSDN博客人人(
(3)方法3:卸载然后重装ubuntu系统
2、vim 找不到命令指令
vim ~/.bashrc
如果在步骤 1 中确认系统缺少vim编辑器,我们需要安装它。使用以下命令来安装vim:
sudo apt-get update
sudo apt-get install vim
3、安装CUDNN
我安装的版本8.9.0
参考这个文章
Ubuntu22.04安装CUDA、cudnn详细步骤_ubuntu安装cuda和cudnn_takedachia的博客-CSDN博客
4、安装Anaconda(已经安装了可跳过)
参考2023年最新Ubuntu安装pytorch教程_七七鸭灬的博客-CSDN博客
或者Ubuntu系统下安装Anaconda3及配置 - 知乎
5、创建anaconda环境
执行下面的命令创建python3.9且名为dl的环境,因为我装的是3.9
conda create -n dl python=3.9
6、安装pytorch
pytorch传送门:Pytorch
安装对应cuda型号的pytorch。具体步骤参考2023年最新Ubuntu安装pytorch教程_七七鸭灬的博客-CSDN博客
7、如何使用spyder 导入pytorch
7.1需要在dl环境里安装spyder
(1)打开spyder
source ~/anaconda3/bin/activate
anaconda-navigator
(2)然后在anaconda图形化界面中选择dl环境,然后install spyder。
也可以用命令行安装:
sudo apt install spyder
7.2每次打开运行spyder
(1)先激活环境
conda activate dl
(2)然后输入就可以打开spyder
spyder
(3)改动spyder中python 编译器,在工具里面,选择一个dl文件里的python
(4)安装第三方库快捷的方法
pip install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple
最核心的是
-i https://pypi.tuna.tsinghua.edu.cn/simple
(Python)使用清华源进行python的pip安装(任何环境,不用换源,用时只需加上一行代码_清华源pip-CSDN博客