Jetson nano，可以在 python3 终端中使用 cuda，但不能在文件中使用

Question

如果我尝试在终端的 jetson nano 上使用 cuda：

$ python3

> > > import torch
> > > print(torch.cuda.is_available())
True

但是，如果我启动一个具有相同内容的文件，则 output 为 False。

有谁知道如何解决这个问题？

我试过了：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

并在 python 终端导入文件。 虽然这会使 cuda 工作，但应用程序的 rest 将无法工作。

我希望在我用python3 file.py启动文件时看到 output True

Answer 1

看来问题与运行文件时未正确设置环境变量有关。 一种解决方案是将以下行添加到脚本的顶部，以确保在导入 torch 之前设置环境变量：

 import os os.environ["CUDA_VISIBLE_DEVICES"] = "0"

这将确保 CUDA 设置为在脚本运行时使用第一个可用设备（设备 0）。 另一种解决方案可能是使用类似的命令

 export CUDA_VISIBLE_DEVICES=0

然后运行您的 python 脚本。 这应该确保在脚本运行时正确设置环境变量。

此外，您还可以检查您使用的 cuda 版本是否与您安装的 Pytorch 版本兼容，并检查您的 jetson nano 是否启用了 cuda GPU。