映象安裝pytorch的簡便方法總結

安全劍客發表於2020-04-04
PyTorch是一個開源的Python機器學習庫,基於Torch,用於自然語言處理等應用程式。2017年1月,由Facebook人工智慧研究院(FAIR)基於Torch推出了PyTorch。本文主要介紹映象安裝pytorch的簡便方法

映象安裝pytorch的簡便方法總結映象安裝pytorch的簡便方法總結

1.安裝anacoda

這網上已有許多教程,這裡不再贅述。

2.顯示卡加速(安裝CPU版本的pytorch跳過)

請原諒我在這裡又省略了,但我找了一個很好的部落格。按照這個來萬無一失,注意安裝時一定要與自己電腦版本匹配。https://zhuanlan.zhihu.com/p/94220564?utm_source=wechat_session

安裝pytorch

進入PyTorch的官網進行版本選擇
https://pytorch.org/get-started/locally/
映象安裝pytorch的簡便方法總結映象安裝pytorch的簡便方法總結
注意這裡package選擇pip
例如:我選擇後的commend為

pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html>

這就是我們的安裝 命令,而在這個安裝 命令之後跟隨映象設定

-i https://pypi.tuna.tsinghua.edu.cn/simple some-package

上面是清華的映象源,現階段已不可使用,提供下面幾個映象源用來參考。
豆瓣(douban) http://pypi.douban.com/simple/

清華大學 https://pypi.tuna.tsinghua.edu.cn/simple/

阿里雲 http://mirrors.aliyun.com/pypi/simple/

中國科技大學 https://pypi.mirrors.ustc.edu.cn/simple/

中國科學技術大學 http://pypi.mirrors.ustc.edu.cn/simple/

清華和科大的映象已經掛掉了,所以題主在這裡使用的是阿里雲的映象,最終的pip命令如下:
pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html -i http://mirrors.aliyun.com/pypi/simple/ some-package
以上,請根據自己選擇的pytorch型號和映象地址改變以上pip命令來滿足您的需要。

原文地址: https://www.linuxprobe.com/methods-installing-pytorch.html

來自 “ ITPUB部落格 ” ,連結:http://blog.itpub.net/31559985/viewspace-2684509/,如需轉載,請註明出處,否則將追究法律責任。

相關文章