这篇文章主要为大家展示了“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进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。