您好,登錄后才能下訂單哦!
這篇文章主要為大家展示了“PyTorch如何安裝使用”,內容簡而易懂,條理清晰,希望能夠幫助大家解決疑惑,下面讓小編帶領大家一起研究并學習一下“PyTorch如何安裝使用”這篇文章吧。
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
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如何安裝使用”這篇文章的所有內容,感謝各位的閱讀!相信大家都有了一定的了解,希望分享的內容對大家有所幫助,如果還想學習更多知識,歡迎關注億速云行業資訊頻道!
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。