Reputation: 770
My apologies as there's a million questions out there like this, but none of them seem to answer mine. I'm trying to re-install Tensorflow so that it uses my GPU.
I'm running:
The CUDA v10.0 folder, the \extras\CUPTI\libx64 and the \include folder are all in my PATH, also CUDAPATH is as well as CUDA_PATH is defined)
Yet, with pip install tensorflow
it only finds my CPU (using:
from tensorflow.python.client import device_lib
device_lib.list_local_devices()
And I can't seem to be able to import tensorflow if I
pip uninstall tensorflow
pip install tensorflow-gpu
python
import tensorflow
ModuleNotFoundError: No module named 'tensorflow'
Upvotes: 1
Views: 2460
Reputation: 420
Try installing Tensorflow again with option --ignore-installed such as:
pip install tensorflow-gpu==2.0.0-alpha0 --ignore-installed
Upvotes: 1