温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

PyTorch如何安装使用

发布时间:2021-11-26 14:55:41 来源:亿速云 阅读:183 作者:小新 栏目:大数据

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

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI