PyTorch是一個開源的Python機器學習庫,基於Torch,用於自然語言處理等應用程序。2017年1月,由Facebook人工智能研究院(FAIR)基於Torch推出了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/
注意這裏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