Skip to content

Commit 9dc4d6a

Browse files
author
Anastasia
committed
Added additional corrections due to the comments
1 parent 958cc1d commit 9dc4d6a

File tree

1 file changed

+1
-11
lines changed

1 file changed

+1
-11
lines changed

KerasCNN-CIFAR/keras-cnn-cifar10.ipynb

Lines changed: 1 addition & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -41,17 +41,7 @@
4141
"tf.random.set_seed(seed)\n",
4242
"np.random.seed(seed)\n",
4343
"\n",
44-
"os.environ['TF_DETERMINISTIC_OPS'] = '1'\n",
45-
"\n",
46-
"\n",
47-
"gpu_devices = tf.config.list_physical_devices('GPU') # get list of GPUs\n",
48-
"\n",
49-
"if gpu_devices: # if there are any gpu devices use first gpu\n",
50-
" tf.config.experimental.set_visible_devices(gpu_devices[0], 'GPU')\n",
51-
" tf.config.experimental.set_memory_growth(gpu_devices[0], True)\n",
52-
" os.environ['TF_USE_CUDNN'] = \"true\"\n",
53-
"else: # else use CPU\n",
54-
" os.environ[\"CUDA_VISIBLE_DEVICES\"] = \"-1\""
44+
"os.environ['TF_DETERMINISTIC_OPS'] = '1'"
5545
]
5646
},
5747
{

0 commit comments

Comments
 (0)