大家好，又见面了，我是你们的朋友全栈君。
Pytorch 安装
已有Cuda 9.0，anaconda3，用conda命令安装pytorch
代码语言：javascript复制conda install pytorch torchvision cuda90 -c pytorch验证是否安装成功
代码语言：javascript复制python然后依次输入
代码语言：javascript复制from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)得到如下之类的输出
代码语言：javascript复制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上是否能用
代码语言：javascript复制import torch
torch.cuda.is_available()
发布者：全栈程序员栈长，转载请注明出处：https://javaforall.cn/152779.html原文链接：https://javaforall.cn
