-
Notifications
You must be signed in to change notification settings - Fork 143
Added Pytorch Cuda Docker Image as the Image pytorch/pytorch:1.0-cuda10.0-cudnn7-runtime in not having cuda so cannot used GPU #248
base: master
Are you sure you want to change the base?
Conversation
Corrected the command to refer to v1 as directory changed from v1beta to v1
Hi @MATRIX4284. Thanks for your PR. I'm waiting for a kubeflow member to verify that this patch is reasonable to test. If it is, they should reply with Once the patch is verified, the new status will be reflected by the I understand the commands that are listed here. Instructions for interacting with me using PR comments are available here. If you have questions or suggestions related to my behavior, please file an issue against the kubernetes/test-infra repository. |
[APPROVALNOTIFIER] This PR is NOT APPROVED This pull-request has been approved by: The full list of commands accepted by this bot can be found here.
Needs approval from an approver in each of these files:
Approvers can indicate their approval by writing |
Can you move the docker file to https://github.com/kubeflow/pytorch-operator/tree/master/examples/mnist and rename it appropriately with mnist changes? |
|
@MATRIX4284 |
I will move it under example folder in a folder named pytorch-gpu.Thanks
for the guidance.
…On Tue, 7 Jan 2020 at 12:15 PM, Johnu George ***@***.***> wrote:
@MATRIX4284 <https://github.com/MATRIX4284>
Thanks for your contribution.
I got your point. However, I feel that it is better not to keep it in the
root folder as it is not related to pytorch operator. Hence I felt, keeping
it in examples looks more appropriate. And users who want to try gpu
version, can refer this example(even if it is a different use case)
—
You are receiving this because you were mentioned.
Reply to this email directly, view it on GitHub
<#248?email_source=notifications&email_token=AE7YQAFGRP4DRJV4EK4WLBLQ4QQHFA5CNFSM4KCO56DKYY3PNVWWK3TUL52HS4DFVREXG43VMVBW63LNMVXHJKTDN5WW2ZLOORPWSZGOEIH4JVY#issuecomment-571458775>,
or unsubscribe
<https://github.com/notifications/unsubscribe-auth/AE7YQAE2NQWGHPWCPJRINQTQ4QQHFANCNFSM4KCO56DA>
.
|
Opened the pr #255 with the docker under the examples folder |
Added Pytorch Cuda Docker Image as the Image pytorch/pytorch:1.0-cuda10.0-cudnn7-runtime in not having cuda.So the examples/mnist.py is not using GPU.The issue is with the pytorch image .The new docker image i supplied is having the cuda dlevel and runtime environment which i tested and working like a breeze on GPU.
The priginal mnist.py which was taking 10 -12 minutes on my double xeon 2670 is taking roughly 1 minute toi get completed using my Titan XP pascal series GPU.
This is the fix to the issue number #245