PyTorch利用GPU加速_torhc如何使用cuda进行tensor计算

作者：小小林熬夜学编程 | 2024-03-18 07:52:34

踩

torhc如何使用cuda进行tensor计算

1.配置cuda版pytorch

登录PyTorch官网（https://PyTorch.org/）,登录后选择对应项
在这里插入图片描述
然后将
复制到命令行，进行安装。
安装之后启动python，执行如下命令，如果没有报错，说明安装成功。

2.pytorch使用GPU计算

查看GPU加速是否可用：

import torch
from torch import nn

print(torch.duda.is_available()) # true 查看GPU是否可用

print(torch.cuda.device_count()) #GPU数量， 1

torch.cuda.current_device() #当前GPU的索引， 0

torch.cuda.get_device_name(0) #输出GPU名称
1
2
3
4
5
6
7
8
9
10

Tensor的GPU计算：

x = torch.ten

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/261483