parent
db3c5b6b54
commit
ea3b2776c4
@ -0,0 +1,2 @@
|
|||||||
|
from google.colab import drive
|
||||||
|
drive.mount('/content/gdrive', force_remount=True)
|
||||||
@ -0,0 +1,5 @@
|
|||||||
|
import tensorflow as tf
|
||||||
|
device_name = tf.test.gpu_device_name()
|
||||||
|
if device_name != '/device:GPU:0':
|
||||||
|
raise SystemError('GPU device not found')
|
||||||
|
print('Found GPU at: {}'.format(device_name))
|
||||||
Loading…
Reference in new issue