报错 OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.

报错OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.
解决：export CUDA_HOME=/usr/local/cuda-X.X