cuda的deriver是全局的,windows和Linux都一样的。
报错原因:是torch版本和cuda版本不匹配。
Step one
cuda是这个网站上下载下来的,
CUDA Toolkit 11.8 Downloads | NVIDIA Developer
这个应该就是cuda,不是驱动,不是torch。
Step twos
进入指令:vim ~/.bashrc
可以看看菜鸟上的Linux vi/vim那一章。Linux vi/vim | 菜鸟教程 (runoob.com)
退出后:source ~/.bashrc
主要是为了修改cuda版本,重点是看11.8的修改区域,之前是的BUG是11.5,不知道为何。
此处是主要修改的关键地方。
(最开始的时候是因为没有装驱动)
1.主要是用命令行去装,没有装驱动,会直接显示的是虚拟机的驱动,可能显示的就是原来Windows的cuda驱动。
CUDA Toolkit 11.8 Downloads | NVIDIA Developer
2.因此driver是不用管的们只需要安装完cuda之后,装cuda换版本就改那个vim ~/.bashrc。
3.nvidia-smi右上角显示的跟自己用的也不是一样的,不必管这个,是全局的cuda版本。
Step three
安装完cuda和修改完环境,到pytorch官网安装torch,
Previous PyTorch Versions | PyTorch
最后的顺序就是如此,先装cuda->改cuda版本->再装pytorch,其中pytorch只能装一个版本,要么就换环境。先改cuda版本再装pytorch,用pip装,cuda你可以装无数个版本,在vim里面改就行了。