Answers for "check tensorflow gpu installation"

5

check if tensorflow gpu is installed

import tensorflow as tf
print(tf.test.gpu_device_name())
Posted by: Guest on August-07-2020
0

install tensorflow gpu

pip3 uninstall tensorflow
pip3 install 'tensorflow-gpu==1.15.4'
Posted by: Guest on August-03-2021

Python Answers by Framework

Browse Popular Code Answers by Language