Yiming Chen
Results
1
comments of
Yiming Chen
可能是pytorch版本不对 首先检查gpu是否可用，运行下面的python代码 ```python import torch device = torch.device("cuda" if torch.cuda.is_available() else "cpu") print("Using device:", device) ``` 如果显示cpu 在终端中输入 ```terminal nvidia-smi ``` 在输出的右上角能找到cuda版本 卸载pytorch，去pytorch官网(https://pytorch.org) 查看对应cuda版本的pytorch并安装