Skip to content

Commit

Permalink
Updated Pytorch and Datascience image changes for ppc64le
Browse files Browse the repository at this point in the history
  • Loading branch information
bhagyashrigai committed Jul 19, 2024
1 parent 40d10b7 commit 9c72aec
Show file tree
Hide file tree
Showing 5 changed files with 1,531 additions and 855 deletions.
5 changes: 5 additions & 0 deletions Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -178,6 +178,11 @@ cuda-jupyter-datascience-ubi9-python-3.9: cuda-jupyter-minimal-ubi9-python-3.9
cuda-jupyter-tensorflow-ubi9-python-3.9: cuda-jupyter-datascience-ubi9-python-3.9
$(call image,$@,jupyter/tensorflow/ubi9-python-3.9,$<)

# Build and push cuda-jupyter-pytorch-ubi9-python-3.9 image to the registry
.PHONY: cuda-jupyter-pytorch-ubi9-python-3.9
cuda-jupyter-pytorch-ubi9-python-3.9: cuda-jupyter-datascience-ubi9-python-3.9
$(call image,$@,jupyter/pytorch/ubi9-python-3.9,$<)

# Build and push jupyter-pytorch-ubi9-python-3.9 image to the registry
.PHONY: jupyter-pytorch-ubi9-python-3.9
jupyter-pytorch-ubi9-python-3.9: jupyter-datascience-ubi9-python-3.9
Expand Down
11 changes: 10 additions & 1 deletion jupyter/datascience/ubi9-python-3.9/Pipfile
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,18 @@ url = "https://pypi.org/simple"
verify_ssl = true
name = "pypi"

[[source]]
url = "http://10.20.177.222:9000/"
verify_ssl = false
name = "localpi"

[dev-packages]

[packages]
ray = {version = "2.7.0.*", index="localpi"}
py-spy = {version = "0.3.14.*", index="localpi"}
pyarrow = {version = "15.0.1.dev0+g5ce6ff434.d20240426", index="localpi"}

# Datascience and useful extensions
boto3 = "~=1.34.50"
kafka-python = "~=2.0.2"
Expand All @@ -17,7 +26,7 @@ scikit-learn = "~=1.4.0"
scipy = "~=1.12.0"
skl2onnx = "~=1.16.0"
onnxconverter-common = "~=1.13.0" # Required for skl2onnx, as upgraded version is not compatible with protobuf
codeflare-sdk = {version = "~=0.16.0", "markers" = "platform_machine != 'ppc64le'"}
codeflare-sdk = "~=0.16.0"

# DB connectors
pymongo = "~=4.6.2"
Expand Down
Loading

0 comments on commit 9c72aec

Please sign in to comment.