-
Notifications
You must be signed in to change notification settings - Fork 283
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
OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root. #19
Comments
The env sets up CUDA_HOME, because pytorch uses that to find the `nvcc`
compiler (which isn't included with conda's installation of cuda, but which
is included in nvidia's standard cuda toolkits). See the following lines
https://github.com/CSAILVision/gandissect/blob/master/script/setup_env.sh#L71-L72
But the choice of CUDA_HOME might be the wrong choice for your system - you
can try editing the same file that is created by that line to change it to
point to the parent directory that contains your installation of nvcc on
your system.
…On Mon, Oct 19, 2020 at 10:13 PM zhiyuan ***@***.***> wrote:
when i create the envs and run the script as mentioned, i always meet the
problem like the title , i have googled the problem with no satisfied
answer, would you mind giving me some suggestions ?
—
You are receiving this because you are subscribed to this thread.
Reply to this email directly, view it on GitHub
<#19>, or unsubscribe
<https://github.com/notifications/unsubscribe-auth/AA2MN2D27V62CW2NQ7FVCGDSLTW4FANCNFSM4SXIKOPQ>
.
|
thanks ,i find the problem may be because of the version of CUDA, when i choose the proper cuda version , the problem is solved |
Is this only for linux? My system is Win, and I input torch.cuda.is_available() in anaconda then I got true. But I still have this trouble. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
when i create the envs and run the script as mentioned, i always meet the problem like the title , i have googled the problem with no satisfied answer, would you mind giving me some suggestions ?
The text was updated successfully, but these errors were encountered: