首先,pip源不信任问题
加比如阿里云的 -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
然后安装pytorch的话需要对应版本,比如我这个服务器的cuda是10.1,我用yolov5需要torch版本>=1.6.0,我决定装1.7.1,那么要去这个pytorch以往版本网站使用相应的命令下载。
然后问题又来了,我cuda10.1版本对应的pip有 +cu101 字样,且后面跟了不同的下载网址,这个在我用的镜像源里是没有的,于是我按照一篇手动安装的文章打开官网给的那个网址,找到cu101、我想安装的torch1.7.1版本、python39、Linux64这个网址,然后在迅雷中新建,输入下载网址,果然速度飞起,700多mb马上下完了,三个whl文件上传到Linux服务器上,打开conda中我的环境,pip安装,马上ok!!!
还有查到很多博主提到,我也可以在Linux中安装很多个cuda版本,这样就不会受限于服务器原本的cuda版本,需要安装cudatoolkit,或许也可行,但还未尝试。
还有一个小问题,vim ~/.condarc 或者bashrc
vim退出要先按Esc,然后输入 :wq (w为保存 q为退出) 回车 就退出了。