when running this in google colab, it no longer works, giving an error when running th

invalid device function running in colab [fix included] about clipdraw HOT 1 OPEN

kvfrans commented on July 16, 2024 2

invalid device function running in colab [fix included]

from clipdraw.

Comments (1)

sousquared commented on July 16, 2024

[Suggestion] Changing the CUDA version

Hi,
Thank you for your great work!
I faced the same error and found that this is because of the default CUDA version of my Colab (CUDA11.1). (The same problem is discussed here.)

!nvcc --version

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:09:46_PDT_2020
Cuda compilation tools, release 11.1, V11.1.105
Build cuda_11.1.TC455_06.29190527_0

So I changed the CUDA version to the one the scripts expected (CUDA10.1 in my case). Then it's working now!

import os
p = os.getenv('PATH')
ld = os.getenv('LD_LIBRARY_PATH')
os.environ['PATH'] = f"/usr/local/cuda-10.1/bin:{p}"
os.environ['LD_LIBRARY_PATH'] = f"/usr/local/cuda-10.1/lib64:{ld}"

!nvcc --version

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243

from clipdraw.

Recommend Projects