189 8069 5689

PyTorch如何安装使用

这篇文章主要为大家展示了“PyTorch如何安装使用”,内容简而易懂,条理清晰,希望能够帮助大家解决疑惑,下面让小编带领大家一起研究并学习一下“PyTorch如何安装使用”这篇文章吧。

海湖新ssl适用于网站、小程序/APP、API接口等需要进行数据传输应用场景,ssl证书未来市场广阔!成为创新互联的ssl证书销售渠道,可以享受市场价格4-6折优惠!如果有意向欢迎电话联系或者加微信:18980820575(备注:SSL证书合作)期待与您的合作!

一、环境介绍

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如何安装使用”这篇文章的所有内容,感谢各位的阅读!相信大家都有了一定的了解,希望分享的内容对大家有所帮助,如果还想学习更多知识,欢迎关注创新互联行业资讯频道!


标题名称:PyTorch如何安装使用
网页URL:http://jkwzsj.com/article/gopgis.html

其他资讯