Если я попытаюсь использовать CUDA на jetson nano в терминале: < /p>
$ python3
> > > import torch
> > > print(torch.cuda.is_available())
True
< /code>
Но если я запускаю файл с тем же контентом, вывод ложный. < /p>
У кого-нибудь есть идеи, как это исправить?import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
< /code>
и импорт файла в терминале Python. Хотя это заставляет CUDA работать, остальная часть приложения не работает.>
Подробнее здесь: https://stackoverflow.com/questions/751 ... -in-a-file
Jetson Nano, может использовать CUDA в терминале Python3, но не в файле ⇐ Python
-
- Похожие темы
- Ответы
- Просмотры
- Последнее сообщение