解决:如何将pytorch的版本改为和cuda对应、如何使用笔记本电脑自带的NVIDIA使用GPU跑深度学习

Step1:安装cuda

网址: https://developer.nvidia.com/cuda-toolkit-archive

PS:此处必须先看看电脑显卡是否自己就装了cuda,可以通过执行命令行语句nvcc -V以此检查cuda是否有,如果有将会显示下面的内容在这里插入图片描述

Step2:执行深度学习命令,检验是否能够跑通

device = 'cuda:0'  # 如果电脑上无Nvidia显卡或未安装CUDA框架,请更改为 'cpu'
x = torch.rand(batch_size, seq_len, input_size).to(device)
rnn = MyRNN(input_size, hidden_size, output_size).to(device)
hidden, y = rnn(x)
print(hidden.shape, y.shape)
  • 如果能够跑通说明可以使用GPU跑深度学习。
  • 如果出现下述报错:
AssertionError: Torch not compiled with CUDA enabled
   该报错表示torch和cuda不兼容,此时只要更换pytorch的版本就可以了,

此时需要转至Step3。

Step3:去往pytorch的官网选择相应出cuda版本并下载pytorch

  • 其中,cuda版本的选择只要低于当前cuda的版本就可以了;

  • 建议不要使用conda安装,因此之后可能会出现由于conda版本未更新等问题而造成安装失败或者安装过慢,因此建议使用pip安装;

  • 使用pip安装时可能会出现超时(即:timeout)等现象,此时可使用管理员的Windows终端执行命令,或者在网络好的时候重试几下等方式解决该问题。博主此处采取的是坚持不懈重试几次解决该问题(一把辛酸泪(╥╯^╰╥)),最终得以成功安装。

以上即为全部内容,OK(*^▽^*)

  • 2
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值