Tensorflow Test File
This file is intended to test the TensorFlow package installation. Successfull execution of the file indicates that you have correctly installed TensorFlow and you are able to use your GPU.
How to use it
-
You can run it by
python tensorflow_test.py
-
Note that the parameter
NUM_ELEMENTS
is used to control the number of loops.- You can reduce its value if you have a memory error
-
There is also a reuiqrement file that you can directly use to setup the environment
- You can do the following steps to setup environment:
- Setup environment:
conda create -n tf_test python=3.7
- Install packages:
conda install --file requirements.txt
- Run file:
python tensorflow_test.py
- Setup environment:
- You can do the following steps to setup environment:
TensorFlow version compatibility
If you are having trouble install the correct version combinations of Tensorflow, cuda, cudnn, here is a super userful image:
And a super useful link which is where the image came from: