如何在RStudio中启用R语言torch包的GPU支持？

问题原因 #

• 直接通过install.packages("torch")安装的是CPU版本的torch包，本身未包含CUDA支持的库，因此无论如何设置CUDA环境变量都无法调用GPU。
• 本地CUDA版本混乱：nvcc显示的编译工具版本为9.1，显卡驱动支持的最高CUDA版本为11.4，你指定的CUDA 10.2既与驱动支持的版本不匹配，也未对应torch的CUDA编译版本。

解决步骤 #

1. 卸载现有CPU版本torch #

remove.packages("torch")

2. 安装带CUDA支持的torch版本 #

先重新安装基础torch包，再用内置函数安装对应CUDA版本的预编译库：

install.packages("torch")
library(torch)
# 可先运行torch::available_cuda_versions()查看torch支持的CUDA版本列表
# 根据你的驱动支持的CUDA 11.4，选择兼容的版本（如11.3）
install_torch(cuda = "11.3")

3. 配置系统CUDA环境变量（可选） #

确保系统环境变量指向与torch匹配的CUDA版本，以CUDA 11.3为例：

export PATH=/usr/local/cuda-11.3/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64:$LD_LIBRARY_PATH

配置完成后重启RStudio。

4. 验证GPU支持 #

运行以下代码测试：

library(torch)
torch_tensor(1, device = "cuda")

若正常返回类似[ CUDAFloatType{1} ]的结果，则说明CUDA配置成功。

注意事项 #

• 无需手动通过Sys.setenv(CUDA="...")指定CUDA路径，torch安装时已绑定对应版本的CUDA库，手动指定会导致冲突。
• 选择CUDA版本时需与显卡驱动兼容：你的驱动版本470.161.03支持最高CUDA 11.4，优先选择11.x系列版本，避免使用10.2这类旧版本。

内容的提问来源于stack exchange，提问作者littleworth