參考:清華源失效後如何安裝pytorch1.01spa
GPU版本的安裝指令爲:.net
1 conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
這裏使用這個命令只是爲了讓Conda自動解決安裝pytorch時候的依賴code