Working with CUDA 10.1 - Githubissues

jackroos / VL-BERT

Code for ICLR 2020 paper "VL-BERT: Pre-training of Generic Visual-Linguistic Representations".

MIT License

738 stars 110 forks source link

Working with CUDA 10.1 #42

Closed clairecyq closed 3 years ago

clairecyq commented 4 years ago

Hi, My gcc and Ubuntu versions match exactly those in README.md, but I have CUDA 10.1. I set up the environment exactly as in README.md. When trying to run the code, I find CUDA 10.1 causing trouble:

if cudatoolkit is not 9.0 -> ImportError: libcudart.so.9.0: cannot open shared object file: No such file or directory
if cudatoolkit is 9.0 -> ImportError:.../VL-BERT/refcoco/../common/lib/roi_pooling/C_ROIPooling.cpython-36m-x86_64-linux-gnu.so: undefined symbol: __cudaRegisterFatBinaryEnd

Is there any fix other than having to install CUDA 9.0?

jackroos commented 4 years ago

You shold install pytorch compiled by cuda 10.1 instead of cuda 9.0 in README like follwing. conda install pytorch=1.1.0 cudatoolkit=10.1 -c pytorch