使用conda或者pip安装的指令

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

测试pytorch_gpu是否可用的代码

# 测试pytorch是否安装成功
import torch
print(torch.__version__)
print(torch.cuda.is_available())
Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐