1.    可以在cmd窗口输入nvidia-smi命令来确认自己的显卡是否有CUDA。

       没有的话，需要安装。（搜CUDA官网按照步骤来）

 2.    然后安装pythorch。

        必须安装的有torch和torchvision两个包，

        ！注意不要直接在Pycharm中自动下载安装，

        否则安装的是CPU版本，无法进行GPU加速。

        上官网：Start Locally | PyTorch   

       选择适合自己电脑的版本。注意是conda安装，不是pip。

       然后打开Pycharm或任意编译器，执行以下代码：

         import torch
         print(torch.cuda.is_available())
         print(torch.__version__)
      

     

         输出 true 即可。（证明可以使用gpu进行加速）

     

  （详情参考：Pytorch使用GPU加速的步骤_pycharm使用gpu加速_微笑小星的博客-CSDN博客）