安装cuda问题

由于ubuntu20.04第一次安装(之前失败过,就放弃了),安装显卡(我的是3090)驱动时我选择软件与更新自动安装了显卡驱动,这也为之后留下了隐患。在之后安装pytorch后,一切程序都可运行,但有一天之前的代码突然不能用了,说是ModuleNotFoundError: No module named 'fused_layer_norm_cuda',然后百度告诉我

是apex安装出了问题
安装apex步骤如下,

git clone https://github.com/NVIDIA/apex.git
cd apex
python setup.py install --cpp_ext --cuda_ext

按照这个步骤,我发现我并不能成功到第三步,后来又百度告诉我

nvcc -V以及torch.version.cuda的版本要一致。好,突然我就去检查发现我运行nvcc-V不行,没有,于是我按照提示的装上还是不行,于是又回到了原点。我突然意识到我的pytorch装成功了吗,我就去检查发现,

import torch    # 如正常则静默
a = torch.Tensor([1.])    # 如正常则静默
a.cuda()

第三句不能正常运行,那可能就是我的cuda版本有问题了,于是我去官网装了cuda10.2,安装步骤如链接 https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal

我发现下载下来后我不能正常安装,提示的错误意思是gcc版本不对,过高(此时是9.3),于是按照这个链接我装了gcc7.5.https://blog.csdn.net/ggggyj/article/details/117691948。

再安装cuda就成功了。

之后就是添加环境了,结束,成功。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值