pytorch-GPU版本安装,避免入坑:
相对来说,安装torch,不论CPU还是GPU都比TensorFlow简单一些,这里讲如何避坑成功安装pytorch-GPU。TensorFlow的GPU版本等我按好了给大家说一下经验。
torch官网安装链接地址https://pytorch.org/get-started/locally/
选择版本:
避坑关键:
推荐使用anaconda的prompt安装方式,其他的本人没试过,感兴趣的同学可以试一试。
其实,很难安装的主要原因是经常需要翻墙,但是网络不稳定,导致超时安装失败,网上有很多同学写的经验,有些比较有用,但是大部分其实不太好用,这里推荐使用阿里云的源文件,你会发现速度一下子快多了,再也不用那么痛苦了。使用就是在上面获取的安装命令后加上如下命令: