pytorch
hailanyi
这个作者很懒,什么都没留下…
展开
-
Ubuntu 安装cuda 11 cudnn8 注意事项,和以前版本安装有所不同
注意和之前版本有所不同1.官网下载cuda 112.安装命令sudo sh cuda_***_linux.run3.设置环境变量vi .bashrc #或者 gedit .bashrc添加以下路径export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/lib64export PATH=$PATH:/usr/local/cuda-11.1/binexport CUDA_HOME=$CUDA_HOME:/usr原创 2021-06-24 22:13:17 · 532 阅读 · 0 评论 -
error: ‘AT_CHECK’ was not declared in this scope解决方法
error: ‘AT_CHECK’ was not declared in this scope在编译deform_conv时遇到问题:error: ‘AT_CHECK’ was not declared in this scope错误原因:AT_CHECK is deprecated in torch 1.5高版本的pytorch不再使用AT_CHECK,而是使用 TORCH_CHECK。解决方法:将所有待编译的源文件中的‘AT_CHECK’全部替换为‘TORCH_CHECK’。...原创 2021-05-21 17:32:36 · 8515 阅读 · 14 评论