当前位置:   article > 正文

【pytorch】安装合集

【pytorch】安装合集

使用conda或者pip安装的指令

https://pytorch.org/get-started/previous-versions/

测试pytorch_gpu是否可用的代码

# 测试pytorch是否安装成功
import torch
print(torch.__version__)
print(torch.cuda.is_available())
  • 1
  • 2
  • 3
  • 4
本文内容由网友自发贡献,转载请注明出处:https://www.wpsshop.cn/w/AllinToyou/article/detail/345524
推荐阅读
相关标签
  

闽ICP备14008679号