解决:如何将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(*^▽^*)

### 解决PyTorch未编译CUDA支持引发的AssertionError 当尝试使用CUDA进行GPU加速时,如果PyTorch版本不是在安装时编译为支持CUDA,则会出现`AssertionError: Torch not compiled with CUDA enabled`错误[^1]。 #### 确认当前环境状态 为了有效解决问题,首先要确认现有的软件硬件条件: - **检查CUDA是否可用** 可通过Python命令来检测CUDA是否已被正确设置: ```python import torch print(torch.cuda.is_available()) ``` 此代码片段用于判断CUDA是否能被PyTorch识别并利用。如果返回False,则意味着存在配置问题[^2]。 - **查询本地CUDA版本** 使用如下命令获取已安装的CUDA版本号以便后续操作中确保各组件间的兼容性: ```bash nvcc --version ``` 该指令能够显示NVIDIA CUDA Compiler (NVCC) 的具体版本信息[^3]。 #### 正确安装带有CUDA支持的PyTorch版本 针对上述情况,推荐卸载现有不带CUDA功能的PyTorch包,并重新安装适合特定操作系统及CUDA版本需求的新版库文件。可以通过pip或conda两种方式完成这一过程;对于大多数用户而言,采用预构建二进制发行版通常是最简便的方法之一[^4]。 例如,要安装适用于CUDA 11.7的PyTorch稳定版,可执行以下pip命令: ```bash pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117 ``` 或者如果是Anaconda环境下,可以考虑使用conda渠道来进行相同的操作: ```bash conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch -c nvidia ``` 以上两条命令分别展示了基于不同工具链下如何指定特定版本CUDA来安装对应PyTorch及其依赖项。 #### 验证新安装的有效性 最后一步是再次运行之前提到过的测试脚本来验证新的安装是否解决了原始的问题。正常情况下,现在应该可以看到True作为输出结果,表明CUDA已经被成功激活并且可供PyTorch调用。 ---
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值