python:windows系统,为什么在jupyter文件里面torch.cuda.is_available()是true,而.py文件里面torch.cuda.is_available()是false?
时间: 2023-08-04 09:08:22 浏览: 170
torch.cuda.is-available()返回False的问题解决
可能是因为在Jupyter Notebook中,你的代码是在Anaconda虚拟环境中运行的,而这个环境中已经安装了CUDA和cuDNN等GPU相关的库,所以torch.cuda.is_available()返回的是True。而在.py文件中,你的代码可能是在另一个环境中运行的,这个环境中没有安装所需要的GPU相关库,因此torch.cuda.is_available()返回的是False。
你可以尝试在.py文件中添加以下代码,来查看你所使用的CUDA版本是否与你所安装的PyTorch版本匹配:
```python
import torch
print(torch.version.cuda)
```
如果输出的CUDA版本与你所安装的PyTorch版本不匹配,则需要更新或安装相应版本的CUDA。
阅读全文