這篇文章主要為大家展示了“PyTorch如何安裝使用”，內(nèi)容簡(jiǎn)而易懂，條理清晰，希望能夠幫助大家解決疑惑，下面讓小編帶領(lǐng)大家一起研究并學(xué)習(xí)一下“PyTorch如何安裝使用”這篇文章吧。

海湖新ssl適用于網(wǎng)站、小程序/APP、API接口等需要進(jìn)行數(shù)據(jù)傳輸應(yīng)用場(chǎng)景，ssl證書(shū)未來(lái)市場(chǎng)廣闊！成為創(chuàng)新互聯(lián)的ssl證書(shū)銷(xiāo)售渠道，可以享受市場(chǎng)價(jià)格4-6折優(yōu)惠！如果有意向歡迎電話聯(lián)系或者加微信：18980820575（備注：SSL證書(shū)合作）期待與您的合作！

一、環(huán)境介紹

cat /etc/*release*
openEuler release 20.03 (LTS)

uname -a
Linux 4.19.90-2003.4.0.0036.oe1.x86_64 #1 SMP Mon Mar 23 19:10:41 UTC 2020 x86_64 x86_64 x86_64 GNU/Linux

nvidia-smi NVIDIA-SMI 440.100 Driver Version: 440.100 CUDA Version: 10.2
GeForce MX250 0MiB / 2002MiB

python3.8 --version
Python 3.8.6

目前torch不支持python3.9

二、安裝

獲取安裝命令： https://pytorch.org/get-started/locally/#linux-pip

PyTorch Build： Stable (1.7.0)
Your OS： Linux
Package： Pip
Language： Python
CUDA： 10.2
Run this Command: pip install torch torchvision

具體命令： pip3.8 install torch torchvision

所有版本下載地址： https://download.pytorch.org/whl/torch_stable.html

三、驗(yàn)證

python3.8 -c "import torch;x = torch.rand(5, 3);print(x)"

tensor([[0.4857, 0.5264, 0.5910], [0.9196, 0.8787, 0.5331], [0.1681, 0.0487, 0.2963], [0.3761, 0.2133, 0.2272], [0.0082, 0.3242, 0.3002]])

python3.8 -c "import torch;print(torch.cuda.is_available())"

True

以上是“PyTorch如何安裝使用”這篇文章的所有內(nèi)容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內(nèi)容對(duì)大家有所幫助，如果還想學(xué)習(xí)更多知識(shí)，歡迎關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道！