最近在服务器上下载了tensorflowgpu后,我准备同时在上面下载pytorch,作为小白还是同时比较这两者的使用情况更能深刻体会二者的不同。
Pytorch下载流程
下载流程(网速好版本)
前言先简单看一下pytorch下载流程:
- 登陆官网,选择合适版本:
Pytorch官网
我根据自己服务器的版本(具体不知道怎么看自己服务器Linux版本的可以看下面这篇文章),选择了:稳定版+Linux+使用pip下载+python+cuda10.2
Linux版本选择(包含tensorflowgpu在Linux上的下载流程)
复制并输入命令如下:
pip install torch torchvision
正常来说这样就可以下载成功,结果如下:
下载流程(网速差版本)
但是凡事哪有那么简单,像我的话就在上面这步代码后报错了,原因是在下载whl文件的时候因为网速太慢耗时太长,所以下到一半就一片触目惊心的红色报错,这时我们可以先从官网下载对应的轮子,然后放到服务器上再下载。torch和torchversion的轮子官网如下