Your exporting code only works for cpu. For gpu, you have to
- change
lib_fname = 'net.o'
to
lib_fname = 'net.tar'
- change
lib.save(lib_path)
to
lib.export_library(lib_path)
Your exporting code only works for cpu. For gpu, you have to
lib_fname = 'net.o'
to
lib_fname = 'net.tar'
lib.save(lib_path)
to
lib.export_library(lib_path)