出现问题的时候,我CUDA版本为9.0,升级到10.0就可以了,下载CUDA10.0安装 CUDA 10.0地址
别忘记安装对应的cuDNN~
如果不想安装新的版本CUDA,可以在程序里把 cuDNN禁用即可,在main函数开始加这么一段代码
torch.backends.cudnn.enabled = False
程序就可以正常运行了,不过就没有了cuDNN的加速,速度会稍微慢一些
出现问题的时候,我CUDA版本为9.0,升级到10.0就可以了,下载CUDA10.0安装 CUDA 10.0地址
别忘记安装对应的cuDNN~
如果不想安装新的版本CUDA,可以在程序里把 cuDNN禁用即可,在main函数开始加这么一段代码
torch.backends.cudnn.enabled = False
程序就可以正常运行了,不过就没有了cuDNN的加速,速度会稍微慢一些