ubuntu 18.04 install pytorch

$ conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

如果無法安裝的話要rm -rf anaconda3，重新安裝anaconda



參考
https://pytorch.org/get-started/locally/