最近GPU 上重新安装pytorch 出现了一些问题,整理一下正确安装pytorch
1.打开pytorch 官网 https://pytorch.org/
如图生成自己安装环境对应的命令 ,如果找不到对应版本,可以直接替换版本号即可,例如在 cuda 9.0 下安装 torch
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch -c defaults -c numba/label/dev
测试安装是否成功 (torch.cuda.is_available())