sm_86 is not compatible with the current PyTorch installation.The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.这就是为什么我升级到最新的CUDA版本,并无法连接到GPU之后。我尝试过很多方法,重新安装库达工具包、PyTorch、torchvision等等,但都没有用。11.6.0_510.39.01_linux.run
$ sudo s
我想训练一个神经网络,我想检查它在每100个时期的总损失,但我在第1行得到错误RuntimeError: CUDA out of memory. Tried to allocate ...,有时我得到这个错误,但这一次因为我尝试了隐藏层中的大量神经元,所以我认为这次这个错误可能是真实的,尽管我尝试了import gc
gc.collect
我想看看PyTorch是否接受了它,所以按照这里的说明:How to check if pytorch is using the GPU?,我运行了以下命令(Python3.6.9,Linux Mint Tricia19.3) >>> import torchTrueKilled
>>> torch.cuda.