Skip to content

Commit fc1763a

Browse files
committed
Added note about TensorFlow CPU container environment
1 parent 728d82a commit fc1763a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

site/devops_pages/tensorflow.md

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,9 @@ Build cuda_12.5.r12.5/compiler.34385749_0
9999

100100
Here is the link to the GitHub repo that will walk you through setting up docker and using the devcontainer: [gperdrizet/tensorflow-GPU](https://github.com/gperdrizet/tensorflow-GPU#).
101101

102-
Target is NVIDIA TensorFlow release 24.06 container. This gets us:
102+
**Note**: there is now a CPU only version of the devcontainer which provides a similar environment for folks without an NVIDIA GPU. You can find it here: [gperdrizet/tensorflow-CPU](https://github.com/gperdrizet/tensorflow-CPU).
103+
104+
For folks with an NVIDIA GPU, the target environment is a NVIDIA TensorFlow release 24.06 container. This gets us:
103105

104106
- CUDA: 12.5
105107
- Python: 3.10

0 commit comments

Comments
 (0)