当前位置: 首页 > 工具软件 > DAIN > 使用案例 >

DAIN配环境问题

章振
2023-12-01

github上说最新适配DAIN的pytorch是1.4.0

按照readme上面requirements安装pytorch1.0.0(https://github.com/baowenbo/DAIN/blob/master/README.md

注释掉my_package/build.sh中source activate pytorch=1.0.0之后install成功

运行后遇到undefined symbol问题

发现setup.py install时,cuda目录是用的服务器usr/local/cuda10.0,但是环境中CUDA=9.0,所以uninstall cudatoolkit9.0,pytorch,torchvision等,重新安装cudatoolkit=10.0等。

现在好像运行成功了

换pytorch版本后重新build.sh

 类似资料: