Add support for V100 GPU. (#940) · a-mp/docker-python@91aeeeb · GitHub

a-mp / docker-python Public

forked from Kaggle/docker-python

• Notifications You must be signed in to change notification settings
• Fork 0
• Star 0

• Code
• Pull requests
• Actions
• Projects
• Security and quality
• Insights

Additional navigation options

Commit 91aeeeb

Browse filesBrowse the repository at this point in the historyBrowse files

rosbo

authored

Add support for V100 GPU. (Kaggle#940)

TensorFlow must be built with 7.0 compute capability. http://b/178104927

1 parent 248f1b3 commit 91aeeeb
Copy full SHA for 91aeeeb

2 files changed

+4-3Lines changed: 4 additions & 3 deletions

File tree

Expand file treeCollapse file tree

Open diff view settings

Filter options

• tensorflow-whl
  • CHANGELOG.md
  • Dockerfile

Expand file treeCollapse file tree

Open diff view settings

Collapse file

‎tensorflow-whl/CHANGELOG.md‎

Copy file name to clipboardExpand all lines: tensorflow-whl/CHANGELOG.md

+2-1Lines changed: 2 additions & 1 deletion

• Display the source diff
• Display the rich diff

Original file line number	Diff line number	Diff line change
@@ -20,4 +20,5 @@
20	20	* `2.3.1-py37`: TensorFlow 2.3.1 with Python 3.7
21	21	* `2.3.1-py37-2`: TensorFlow 2.3.1 & TFA 0.11 with Python 3.7
22	22	* `2.4.0-py37`: TensorFlow 2.4.0 & Python 3.7 & TFA 0.12
23		- * `2.4.0-py37-2`: TensorFlow 2.4.0 & m61 base image
	23	+ * `2.4.0-py37-2`: TensorFlow 2.4.0 & m61 base image
	24	+ * `2.4.0-py37-3`: TensorFlow 2.4.0 & 7.0 CUDA compute capability

Collapse file

‎tensorflow-whl/Dockerfile‎

Copy file name to clipboardExpand all lines: tensorflow-whl/Dockerfile

+2-2Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
@@ -103,8 +103,8 @@ RUN cd /usr/local/src/tensorflow_gcs_config && \
103	103	# Create a tensorflow wheel for GPU/cuda
104	104	ENV TF_NEED_CUDA=1
105	105	ENV TF_CUDA_VERSION=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION
106		- # 3.7 is for the K80 and 6.0 is for the P100, 7.5 is for the T4: https://developer.nvidia.com/cuda-gpus
107		- ENV TF_CUDA_COMPUTE_CAPABILITIES=3.7,6.0,7.5
	106	+ # 3.7 (K80), 6.0 (P100), 7.0 (V100), 7.5 (T4): https://developer.nvidia.com/cuda-gpus
	107	+ ENV TF_CUDA_COMPUTE_CAPABILITIES=3.7,6.0,7.0,7.5
108	108	ENV TF_CUDNN_VERSION=7
109	109	ENV TF_NCCL_VERSION=2
110	110	ENV NCCL_INSTALL_PATH=/usr/

0 commit comments

Comments

0 (0)