PowerShell 中的 Mozilla TTS:“用户警告:具有 CUDA 功能的 NVIDIA GeForce RTX 3060 Ti sm_86 与当前的 PyTorch 安装不兼容
Mozilla TTS in PowerShell: "UserWarning: NVIDIA GeForce RTX 3060 Ti with CUDA capability sm_86 is not compatible with the current PyTorch installation
我正在尝试 运行./TTS/bin/train_tacotron.py 在 Powershell 中使用 GPU。
我遵循了这些 ,这让我走得很远:配置已读取,模型已恢复,但随着训练即将开始,我收到消息:
UserWarning: NVIDIA GeForce RTX 3060 Ti with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
If you want to use the NVIDIA GeForce RTX 3060 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
指定的说明并没有真正帮助。我尝试安装 PyTorch 的最新稳定版本,并尝试安装 1.7.1(而不是我链接的说明中推荐的 1.8.0),但我收到了相同的消息。
我怎样才能在我的 GPU 上达到 运行?
旁注:我能够在 WSL 中成功地 运行 在我的 GPU 上进行训练,但它在几百个 epoch 后就冻结了,所以我想尝试 Powershell 看看它是否有所作为。
为了与您当前的 CUDA 版本一起正常工作,您需要为 cudatoolkit 指定版本 11.3。执行以下命令:
conda uninstall cudatoolkit
conda install cudatoolkit=11.3 -c pytorch
我正在尝试 运行./TTS/bin/train_tacotron.py 在 Powershell 中使用 GPU。
我遵循了这些
UserWarning: NVIDIA GeForce RTX 3060 Ti with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37. If you want to use the NVIDIA GeForce RTX 3060 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
指定的说明并没有真正帮助。我尝试安装 PyTorch 的最新稳定版本,并尝试安装 1.7.1(而不是我链接的说明中推荐的 1.8.0),但我收到了相同的消息。
我怎样才能在我的 GPU 上达到 运行?
旁注:我能够在 WSL 中成功地 运行 在我的 GPU 上进行训练,但它在几百个 epoch 后就冻结了,所以我想尝试 Powershell 看看它是否有所作为。
为了与您当前的 CUDA 版本一起正常工作,您需要为 cudatoolkit 指定版本 11.3。执行以下命令:
conda uninstall cudatoolkit
conda install cudatoolkit=11.3 -c pytorch