r 在火炬中不可用 cuda

Posted 2023-03-29

【中文标题】r 在火炬中不可用 cuda

【英文标题】：cuda unavailable in torch with r

【发布时间】：2021-06-24 16:08:01

【问题描述】：

我在 Python 环境中使用过深度学习。另外，我可以在 r 中使用带有 gpu 的 keras。但是，torch（r 包中的其他深度学习框架）不能使用 gpu。当我用谷歌搜索时，CUDA 版本可能会出现问题。所以我下载了 10.2v 和最新版本 11.2v。然后，它进入路径。

Sys.getenv()
CUDA                                      10.2
CUDA_PATH                                 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2
CUDA_PATH_V10_0                           C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
CUDA_PATH_V10_2                           C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2
CUDA_PATH_V11_2                           C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2

但是

>cuda_is_available()
FALSE

我怎样才能固定返回 TRUE？

【问题讨论】：

我试过最新的 cuda 版本，但还是一样。

【参考方案1】：

在我的情况下，第二行源代码出现了错误代码。但是只需重新安装，您就可以在 Tor 中使用 gpu 和 R！

Sys.setenv("CUDA_HOME" = "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.2") 
source("https://raw.githubusercontent.com/mlverse/torch/master/R/install.R")    
install.packages("torch")