首页 > 其他分享 >win10 torchvision训练模型错误

win10 torchvision训练模型错误

时间:2022-11-25 21:36:25浏览次数:72  
标签:torchvision 模型 torchvison 编译 pytorch win10 cuda10.2

pytorch-1.8.2+cuda10.2+torchvison-0.9.2(有问题)

pytorch/vision链接

1、拉vision代码

1)代码版本号需要跟pytorch对应上,也可以跟torchvison版本一致

2、重新编译生成torchvision:

1)如果直接使用安装的torchvision,会报以下错误:

RuntimeError: Couldn't load custom C++ ops. This can hap......

2)编译安装torchvision

(1)采用默认的visual studio 2019编译生成,好像对vs版本没啥要求!

(2)也会调用本机环境中的c++版的cuda10.2

python setup.py build develop

 

标签:torchvision,模型,torchvison,编译,pytorch,win10,cuda10.2
From: https://www.cnblogs.com/wllwqdeai/p/16926399.html

相关文章