检查R语言的torch是否支持GPU

library(torch)  
if (cuda_is_available()) {  
  cat("CUDA is available\n")  
} else {  
  cat("CUDA is not available\n")  
}  

或者

torch::cuda_is_available()

评论

发表回复