设置
安装此仓库时,建议创建一个新的 conda 环境。为您的设置安装 PyTorch GPU/CPU。
conda create -n mvenv python=3.7
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
使用以下命令运行 setup.py 文件,它将安装 requirements.txt
中列出的所有包。
pip install --upgrade pip setuptools wheel
pip install -e .
验证设置,您可以运行测试
cd tests
coverage run -m unittest discover -v
测试覆盖率可以通过 coverage report
查看。