Hugging Face's logo

Spaces:
vrodriguezf
/
deepvats
Sleeping

App Files Community
2
misantamaria commited on
Commit
285d9e7
·
1 Parent(s): ba2fe7f

Corregido set_device a 0

Browse files
Files changed (1) hide show
  1. r_shiny_app/global.R +1 -1
r_shiny_app/global.R CHANGED
@@ -32,7 +32,7 @@ torch <- reticulate::import("torch")
32
  #options(shiny.trace = TRUE)
33
  if(torch$cuda$is_available()){
34
  print(paste0("CUDA AVAILABLE. Num devices: ", torch$cuda$device_count()))
35
- torch$cuda$set_device(as.integer(1))
36
  #torch$cuda$set_device(as.integer(1))
37
  #torch$cuda$set_device(as.integer(2))
38
  #print(torch$cuda$memory_summary())
 
32
  #options(shiny.trace = TRUE)
33
  if(torch$cuda$is_available()){
34
  print(paste0("CUDA AVAILABLE. Num devices: ", torch$cuda$device_count()))
35
+ torch$cuda$set_device(as.integer(0))
36
  #torch$cuda$set_device(as.integer(1))
37
  #torch$cuda$set_device(as.integer(2))
38
  #print(torch$cuda$memory_summary())