怎么在anaconda中显示pytorch是cup版本还是gpu版本
时间: 2023-06-20 08:10:21 浏览: 980
anaconda配置pytorch环境-Anaconda环境中PyTorch配置详解
在anaconda中,可以通过以下命令查看pytorch是否是GPU版本:
```
import torch
print(torch.version.cuda)
```
如果输出结果为一个数字,则表示安装的是GPU版本的PyTorch,该数字代表CUDA的版本号。如果输出结果为None,则表示安装的是CPU版本的PyTorch。
阅读全文