Получить путь среды CUDA_HOME PYTORCH - PullRequest
Новая
       15

Получить путь среды CUDA_HOME PYTORCH

0 голосов
/ 10 октября 2018

У меня установлена ​​система cuda через anaconda в моей системе с двумя графическими процессорами, которая распознается моим питоном.

import torch
torch.cuda.is_available()
true

Однако, когда я пытаюсь запустить модель через ее C API, я получаю следующую ошибку: 

~/anaconda3/lib/python3.6/site-packages/torch/utils/cpp_extension.py in _join_cuda_home(*paths)
722     '''
723     if CUDA_HOME is None:
--> 724         raise EnvironmentError('CUDA_HOME environment variable is not set. '
725                                'Please set it to your CUDA install root.')
726     return os.path.join(CUDA_HOME, *paths)

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

https://lfd.readthedocs.io/en/latest/install_gpu.html страница дает инструкцию по настройке CUDA_HOMEпуть, если cuda установлен через их метод.Так как я установил cuda через anaconda, я не знаю, какой путь установить.

Я попытался найти метод, но он возвращает мне слишком много путей для cuda.Может кто-нибудь помочь мне с путем для CUDA.Заранее спасибо.

...