Open Wei-Peng-fei opened 1 year ago
Same error . Have you solved this problem?
You might need to write in your code this line. _os.environ['CUDAHOME']="/usr/local/cuda-11.8"
Change the string to you cuda path and It might work.
You might need to write in your code this line. _os.environ['CUDAHOME']="/usr/local/cuda-11.8"
Change the string to you cuda path and It might work.
where to write the this line of code pls
At the beginning of the code. You need to import the os module first.
import os
os.environ['CUDA_HOME']="/usr/local/cuda-11.8"
I wrote it in the Tracker.py file, but it should work in any other file as long as you access that file. You can also do
export CUDA_HOME=/usr/local/cuda-11.8
but you will need to do it every time you access the CMD to execute the file.