Windows 10 python3.6安裝pytorch


  近年來,深度學習框架如雨后春筍般的涌現出來,如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

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM