pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp35-cp35m-linux_x86_64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple docker-compose --user
注:cu表示的是cuda的版本,cp表示的是python的版本,后面-i是下載的清華鏡像
pip3 install torchvision==0.2.2 --user
舊版本的0.4.1的torch對應的torchvision是0.2.2的,要安裝對應的版本
寫的是對應的本地的安裝方式,有sudo 權限的用sudo pip,去掉--user