跳到内容

设置

安装此仓库时,建议创建一个新的 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 查看。