windows上已经安装过显卡驱动,可在终端通过nvcc -V命令查看
在代码中检查cuda是否启用:
print(torch.cuda.is_available())
如果输出的结果是False,那么说明当前的Pytorch版本无法使用显卡
需要重新安装GPU版的Pytorch
可参考以下博客的安装过程
windows10 Anaconda安装pytorch(GPU)踩坑!
windows上已经安装过显卡驱动,可在终端通过nvcc -V命令查看
在代码中检查cuda是否启用:
print(torch.cuda.is_available())
如果输出的结果是False,那么说明当前的Pytorch版本无法使用显卡
需要重新安装GPU版的Pytorch
可参考以下博客的安装过程
windows10 Anaconda安装pytorch(GPU)踩坑!