最近在服务器上下载了tensorflowgpu后,我准备同时在上面下载pytorch,作为小白还是同时比较这两者的使用情况更能深刻体会二者的不同。
Pytorch下载流程
下载流程(网速好版本)
前言先简单看一下pytorch下载流程:
- 登陆官网,选择合适版本:
Pytorch官网
我根据自己服务器的版本(具体不知道怎么看自己服务器Linux版本的可以看下面这篇文章),选择了:稳定版+Linux+使用pip下载+python+cuda10.2
Linux版本选择(包含tensorflowgpu在Linux上的下载流程)
复制并输入命令如下:
pip install torch torchvision
正常来说这样就可以下载成功,结果如下: