Pytorch version on NPU in VIM3

VIM3
1

请问贵公司，我现在在VIM3主板上烧录了Ubuntu系统并安装了pycharm及python3.7，想把原笔记本上面的程序测试一下，但是不知道pytorch该下载哪个版本。Pytorch官网上中的Compute Platform包括CUDA、ROCm和CPU，是不是应该选择ROCm？

2

@ShaojinMa 应该是CPU的版本

3

谢谢您。那我是在python里面能够调用吗？是通过某句话调用npu运行？比如调用gpu时候：device = torch.device(“cuda:0” if torch.cuda.is_available() else “cpu”)，然后，net.to(device)就可以了

4

@ShaojinMa Python的NPU测试版本即将发布，到时候你可以尝试一下Python的

1 Like
5

明白了。那再问您一下，我现在已经写好了网络模型，有其他方法调用吗？都是.py文件

6

@ShaojinMa 什么意思，写好了网络模型？你直接讲其他平台的代码拿过来，是只能在CPU上面跑，用不了NPU的

7

是自己写的网络，已经保存成.pkl格式的网络模型，能用NPU调用吗？

8

@ShaojinMa 不可以，你要调用npu，必须使用sdk去转换。原始的模型在vim3山面跑就是跑的cpu，效率跟PC上的cpu是没办法比较的