解决某些深度学习代码使用了cpp_extension,CUDAExtension编译了代码可能存在的运行环境版本过低问题,进行cuda和pytorch的运行环境升级
综述--------------------------------------------------------------------1. 问题使用Transformer中PNS-Net代码 ( 论文 | 数据集及代码) ,其要求环境是python3.6,pytorch=1.1.0,重点来了,对服务器环境外的要求是这样的:·Mark 1 ↓Our core design is built on CUDA OP with torchlib. Please ensure the base C
原创
2022-03-10 23:20:11 ·
4760 阅读 ·
0 评论