记录一下跑BARF训练的过程

环境

Ubuntu18.04+anaconda3
按照BARF的readme配置环境（后续给出网址）

错误

测试时发现torch.cuda.is_available()返回false，网上搜索发现有以下几个问题：
1.显卡驱动版本，cuda版本和torch版本要对应
2.conda配置的torch版本似乎无效，需要卸载重装

解决

按照网上教程安装了显卡驱动，我的版本是510.47.03

这里显示cuda推荐版本是11.6，于是在官网按照教程安装了cuda11.6
在conda环境下卸载torch和torchvision，并根据官网教材安装torch和torchvision，由于目前只支持到cuda11.3，搜索发现也可用于cuda11.6。

pip uninstall torch
pip uninstall torchvision
pip3 install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio==0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

结果

可以发现这里打印输出True

参考

1. https://blog.youkuaiyun.com/bird11111/article/details/105770644
2. https://blog.youkuaiyun.com/bird11111/article/details/105746686