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
[Caffe2] cudnn mismatch #17800
Comments
I guess that you'll need cudatoolkit-dev for building PyTorch. |
|
conda install -c conda-forge cudatoolkit-dev |
Sorry, it was late yesterday so I didn't have too much time to explain it clearly. |
Thanks for your patience and help ! |
cc @kostmo |
❓ Questions and Help
Hi, Everyone.
This is my system config:
cuda: 8.0 cudnn: 6.0.21
Environment : A virtual environment, with
python2.7 in Anaconda2, cudatoolkit: 8.0, cudnn: 7.1.3
When I install caffe2 from source code, and run codes based on it, i met following issue:
And When I build, I check the cuda and cudnn information like this:
And i met these warning infos at the last:
This is the command I used in cmake:
cmake .. -DCMAKE_PREFIX_PATH=/home/pci/anaconda2/envs/spotter -DCMAKE_INSTALL_PREFIX=/home/pci/anaconda2/envs/spotter -DPYTHON_LIBRARY=/home/pci/anaconda2/envs/spotter/lib/python2.7/site-packages -DPYTHON_INCLUDE_DIR=/home/pci/anaconda2/envs/spotter/include/python2.7
-DUSE_NCCL=OFF
It shows that caffe2 builded with cuda8 and cudnn7.1 in build information, But...
So Why this happend...? And I want to know how to fix it...
Thank you very much !
The text was updated successfully, but these errors were encountered: