安装gpu-pytorch版本

0.安装的前提是conda安装网络是通的，详情见https://www.cnblogs.com/pyclq/p/15528029.html
1、安装GPU 版本的pytorch核心要点是 a.根据 cude驱动安装合适的 cudatoolkit版本 b.根据合适的 cudatoolkit版本安装合适的pytorch版本
* 查看 nvidia-smi驱动版本

* 根据驱动版本，确定合适的cudatoolkit和pytorch版本，对应关系可以参考下面链接

https://blog.csdn.net/weixin_42069606/article/details/105198845?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.channel_param&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.channel_param

* 安装cudatoolkit和pytorch
      conda install cudatoolkit=10.1.243
      conda install pytorch==1.6 python==3.6.5 -c pytorch

       conda install pytorch==1.6 cudatoolkit=10.1.243 python==3.6.5 --no-update-dependencies            (需要加上--no-update-dependencies 否则python版本会换，进而导致原conda环境被换)  
* 进入python测试cuda是否可用
  import torch
  torch.cuda.is_available()