grosse
/
test-ci
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

try without limit
ci/woodpecker/push/woodpecker Pipeline was successful Details

This commit is contained in:
Malte Grosse 2024-04-11 16:44:03 +09:00
parent b3fdc6dcf6
commit 67f6ecbbb9
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
run.py
View File

@ -9,12 +9,12 @@ from keras import backend as K
K.clear_session() K.clear_session()
gpus = tf.config.experimental.list_physical_devices('GPU') gpus = tf.config.experimental.list_physical_devices('GPU')
if gpus: # if gpus:
try: # try:
tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=6024)]) # tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=6024)])
except RuntimeError as e: # except RuntimeError as e:
print(e) # print(e)
os.exit(1) # os.exit(1)
print(tf.config.experimental.list_physical_devices()) print(tf.config.experimental.list_physical_devices())