原本安裝pytorch只需要一條命令:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
結果不是網不行就是各種出錯。。。
於是用pip安裝試了一下,發現還是pip好用啊,conda說的那麼好。。。
這是官網推薦的
pip install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl
pip install https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp36-cp36m-win_amd64.whl