Ubuntu下配置Pytorch环境踩坑-指定安装源和Build环境安装Pytorch

问题:最近在服务器上部署YoloX,前辈配置的虚拟机上只配置了DeepStream、TensorRT、CUDA这些,所以打算配置一下PyTorch环境,最头疼的问题应该是torch的下载。输入官网的从官方源的下载命令后,果然torch下载没有速度,由于不会用ubuntu挂梯子,就在笔记本电脑上下载,再挪到服务器上,但是安装时报错说找不到用于配置的轮子,仔细检查后发现,在笔记本电脑上(windows环境)下载的是 win环境下的安装包,在清华源上果然发现了linux的安装包,于是通过清华源下载对应版本。

指定源下载是通过pip3 install pytorch=version=build_env -c 清华源仓库地址的方式,这样能够指定源、安装包版本、build环境。

posted @ 2022-05-23 20:24  Sylphh  阅读(13)  评论(0)    收藏  举报