Reputation: 21
I want to do some ML on my computer with Python, I'm facing problem with the installation of tensorflow and I found that tensorflow could work with GPU, which is CUDA enabled. I've got a GPU Geforce gtx 1650, will tensorflow work on that.
If yes, then, how could I do so?
Upvotes: 1
Views: 3472
Reputation: 304
I don't think if you can. https://www.tensorflow.org/install/gpu Tensorflow clearly mentions the list of supported architectures and the 1650 sadly doesn't belong to the list. Check the "cuda enabled gpu cards" link on the website above.
Upvotes: 0
Reputation: 2085
Here are the steps for installation of tensorflow:
pip install tensorflow
Upvotes: 0
Reputation: 119
After opening the command prompt in administrator mode,the installation command for Tensorflow with GPU support is as follows:
pip3 install --upgrade tensorflow-gpu
To check if tensorflow has been successfully installed use command:
import tensorflow as tf
To test CUDA support for your Tensorflow installation, you can run the following command in the shell:
tf.test.is_built_with_cuda()
[Warning: if a non-GPU version of the package is installed, the function would also return False. Use this command to validate if TensorFlow was build with CUDA support.]
Finally, to confirm that the GPU is available to Tensorflow, you can test using a built-in utility function in TensorFlow as shown below:
tf.test.is_gpu_available(cuda_only=False, min_cuda_compute_capability=None)
Upvotes: 2
Reputation: 51
Install tensorflow-gpu to do computations on GPU. You can use the code below to check whether your GPU is being used by tensorflow.
tf.test.is_gpu_available(
cuda_only=False,
min_cuda_compute_capability=None
)
Upvotes: 1