• PyTorch官网：https://pytorch.org/get-started/locally/

使用Pip安装CUDA11.6版的PyTorch的命令如下：

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

注意：如果之前安装过CPU版本的，可以先卸载一下，卸载命令如下：

pip uninstall torch -y
pip uninstall torchvision -y

如何安装 NVIDIA 驱动程序可以参考：https://helpx.adobe.com/cn/premiere-pro/kb/drivers-video-win-nvidia.html