Pytorch 安装

Pytorch 安装Pytorch安装已有Cuda9.0,anaconda3,用conda命令安装pytorchcondainstallpytorchtorchvisioncuda90-cpytorch验证是否安装成功python然后依次输入from__future__importprint_functionimporttorchx=torch.rand(5,3)p…

大家好,又见面了,我是你们的朋友全栈君。

Pytorch 安装

已有Cuda 9.0,anaconda3,用conda命令安装pytorch

conda install pytorch torchvision cuda90 -c pytorch

验证是否安装成功

python

然后依次输入

from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)

得到如下之类的输出

0.1820  0.1333  0.2273
 0.9710  0.9239  0.4220
 0.5378  0.4325  0.6748
 0.1138  0.1160  0.1531
 0.3309  0.7117  0.6854
[torch.FloatTensor of size 5x3]

验证pytorch在当前GPU和cuda上是否能用

import torch
torch.cuda.is_available()

在这里插入图片描述

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请联系我们举报,一经查实,本站将立刻删除。

发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/152779.html原文链接:https://javaforall.net

(0)
全栈程序员-站长的头像全栈程序员-站长


相关推荐

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注全栈程序员社区公众号