记录cuda和pytorch在wsl中的安装经验(shout out to 朱大佬)

文章讲述了CUDA驱动与torch版本不匹配的问题,以及解决步骤。首先从NVIDIA官网下载CUDAToolkit,然后通过修改Linux的.bashrc文件来切换CUDA版本。强调driver无需特别处理,安装CUDA后,根据系统环境安装对应版本的PyTorch,顺序为:安装CUDA->修改CUDA版本->安装PyTorch。
摘要由CSDN通过智能技术生成

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里面改就行了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值