近年來,深度學習框架如雨后春筍般的涌現出來,如TensorFlow、caffe、caffe2、PyTorch、Keras、Theano、Torch等,對於從事計算機視覺/機器學習/圖像處理方面的研究者或者教育者提高了更高的要求。其中Pytorch是Torch的升級版,其有非常優秀的前端和靈活性,相比TensorFlow不用重復造輪子,易於Debug調試,極大的提高開發效率,使得其在其他框架中脫穎而出。更多信息參見:caffe2+Pytorch1.0 = Pytorch1.0,期待即將推出的1.0版本,caffe2更適合后端部署。
1.進入pytorch官網https://pytorch.org/
選擇對應的版本、環境、安裝方式等:
將生成的命令在終端進行輸入,安裝成功。
測試:
import torch#導入沒有報錯說明安裝成功
print(torch.cuda.is_available())#查看是否支持GPU