VMs need CUDA 11 for tf-nightly releases >= 2.4.0-dev20200819 to see the GPU
See original GitHub issueBug report for Colab: http://colab.research.google.com/.
-
Describe the current behavior: As of
tf-nightly
release2.4.0-dev20200819
the GPU devices are not detected when connected to GPU. As mentioned here, that release is the first one to use CUDA 11 instead of CUDA 10.1. -
Describe the expected behavior: The GPU devices should be detected when the accelerator is set to ‘GPU’.
-
The web browser you are using (Chrome, Firefox, Safari, etc.): Chrome
-
Link (not screenshot!) to a minimal, public, self-contained notebook that reproduces this issue (click the Share button, then Get Shareable Link): https://colab.research.google.com/gist/amahendrakar/25b8598fea215239f22d406a387c1735/42957.ipynb
Issue Analytics
- State:
- Created 3 years ago
- Reactions:11
- Comments:13 (3 by maintainers)
Top Results From Across the Web
Tensorflow 2.4 with CUDA 11.2 -GPU training fix - Medium
I was using TensorFlow 2.3.1. Step 1: Review GPU utilization. This is a basic step to find out ...
Read more >CUDA Compatibility :: NVIDIA Data Center GPU Driver ...
In order to run a CUDA application, the system should have a CUDA enabled GPU and an NVIDIA display driver that is compatible...
Read more >No gpu support with tf-nightly 2.4.0 and I get "Could not load ...
For the newest version of tf-nightly you should be using cuda 11.0, if you install it alongside cuDNN 8.0, your problems should disappear....
Read more >Install TensorFlow with pip
For the preview build (nightly), use the pip package named tf-nightly . ... Note: GPU support is available for Ubuntu and Windows with...
Read more >CUDA 11 Perf regression - Google Groups
Tensorflow Pip version. Cuda version. cudnn version. Images/sec. tf-nightly (2.4.0-dev20201007). 11.0.3. 8.0.4. 8635.73. tf-2.3.
Read more >
Top Related Medium Post
No results found
Top Related StackOverflow Question
No results found
Troubleshoot Live Code
Lightrun enables developers to add logs, metrics and snapshots to live code - no restarts or redeploys required.
Start Free
Top Related Reddit Thread
No results found
Top Related Hackernoon Post
No results found
Top Related Tweet
No results found
Top Related Dev.to Post
No results found
Top Related Hashnode Post
No results found
CUDA 11.0 / CUDNN 8 is now available in Colab VMs, and tf-nightly (tf-nightly-2.5.0.dev20210302) can see the GPU:
Is it possible to install a tf-nightly version that use CUDA 10.1 as August builds have been removed in pip ?