Вычисления на GPU в Google Colabaratory - PullRequest
Новая
       58

Вычисления на GPU в Google Colabaratory

0 голосов
/ 08 мая 2020

У меня есть ноутбук в G C с настроенными вычислениями gpu. Когда я запускаю: 

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

, я вижу GPU на устройствах: 

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 2881683908768157563
, name: "/device:XLA_CPU:0"
device_type: "XLA_CPU"
memory_limit: 17179869184
locality {
}
incarnation: 14433918882325490908
physical_device_desc: "device: XLA_CPU device"
, name: "/device:XLA_GPU:0"
device_type: "XLA_GPU"
memory_limit: 17179869184
locality {
}
incarnation: 8857856280193037152
physical_device_desc: "device: XLA_GPU device"
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 15701463552
locality {
  bus_id: 1
  links {
  }
}
incarnation: 13142570581108506915
physical_device_desc: "device: 0, name: Tesla P100-PCIE-16GB, pci bus id: 0000:00:04.0, compute capability: 6.0"
]

, но когда я помещаю этот код в файл .py, который я запускаю из оболочки, используя 

!python3 file.py

Не вижу GPU в устройствах: 

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 5088676628756383267
, name: "/device:XLA_CPU:0"
device_type: "XLA_CPU"
memory_limit: 17179869184
locality {
}
incarnation: 550355900248427842
physical_device_desc: "device: XLA_CPU device"
]

Что это такое и как исправить?

...