【问题标题】:cuda unavailable in torch with rr 在火炬中不可用 cuda
【发布时间】: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 版本,但还是一样。

标签: r gpu torch


【解决方案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")

【讨论】:

    猜你喜欢
    • 2019-08-17
    • 2017-06-06
    • 2017-02-23
    • 2020-07-06
    • 2022-07-20
    • 2018-09-02
    • 1970-01-01
    • 2017-04-09
    • 2021-03-30
    相关资源
    最近更新 更多