主机下torch.cuda.is_available()=true虚拟环境下torch.cuda.is_available()=false
时间: 2024-04-22 12:20:22 浏览: 96
torch.cuda.is-available()返回False的问题解决
在主机环境下,当调用`torch.cuda.is_available()`函数时,如果返回为`True`,表示主机上存在可用的CUDA设备,可以使用GPU进行计。而在虚拟环境下,特别是在没有正确配置虚拟环境的情况下,调用`torch.cuda.is_available()`函数可能会返回`False`,表示虚拟环境中没有可用的CUDA设备。
阅读全文