-
Notifications
You must be signed in to change notification settings - Fork 97
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
run without CUDA #7
Comments
Simply removing the two calls to .cuda in |
That does not work. We did try that.
We get an error saying no NVIDIA device found.
We would like to replicate your results. Is it possible for you to commit 2 new working codes of |
with
|
with
|
please do note that we have imported the following model for |
The torchvision net is not quite a drop-in replacement. Get the git submodule for the caffe resnet fixed and try the pin_memory=False version. Either way, I don't recommend running this with a CPU-only -- it will take ages to train. |
Okay. Thanks
…On 5 January 2018 at 18:49, Yan Zhang ***@***.***> wrote:
The torchvision net is not quite a drop-in replacement. Get the git
submodule for the caffe resnet fixed and try the pin_memory=False version.
Either way, I don't recommend running this with a CPU-only -- it will take
ages to train.
—
You are receiving this because you authored the thread.
Reply to this email directly, view it on GitHub
<#7 (comment)>,
or mute the thread
<https://github.com/notifications/unsubscribe-auth/AY_7IZrqCg9fK09ttXvHkiADB01wV7Wmks5tHiFLgaJpZM4RSnZB>
.
|
Is there a way to convert the preprocess-images.py to a version that doesnt require CUDA?
The text was updated successfully, but these errors were encountered: