解决Pytorch中Cuda无法GPU加速问题_北海虽赊,扶摇可接的博客-CSDN博客_pytorch无法使用gpu
我安装好了CUDA 11.3 并且也安装好了对应的cuDNN,创建了一个虚拟环境安装了GPU版的pytorch,但是安装好后发现pytorch无法进行GPU加速。按照这个教程通过以下代码就能解决这个问题。
pip3 install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio===0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
pip3 install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio===0.10.2+cu113
只需要按照这一部分的内容进行解决就可以了