美文网首页
3、深度学习库Pytorch

3、深度学习库Pytorch

作者: 木火_magic | 来源:发表于2021-11-17 01:33 被阅读0次

*Pyorch官网

*Pytorch安装指南

安装Pytorch的pip命令

pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio==0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

对应的库安装完,可以用python代码验证PyTorch是否安装成功

import torch
torch.cuda.is_available()
要注意的是,输出结果为True或者False 仅代表GPU是否可用,如无报错,均说明PyTorch库安装成功

相关文章

网友评论

      本文标题:3、深度学习库Pytorch

      本文链接:https://www.haomeiwen.com/subject/pegktrtx.html