问题描述
最近使用rppg-toolbox来训练一些模型,直接按照rppg-toolbox主页上的安装方式安装,在A6000主机上使用torch时产生了warning,并且程序运行十分缓慢
原因
产生这个问题的原因是因为按照rppg-toolbox的安装方式,安装的pytorch1.12.1是不支持CUDA11.4的,使用PiP list查看当前安装的pytorch版本:
而现在使用的GPU驱动、cuda版本为:
解决方案
为了解决这个问题,我们需要手动安装rppg-toolbox,首先运行命令删除之前的环境
conda remove --name rppg-toolbox --all -y
删除原来的环境紧接着创建新环境
conda create -n rppg-toolbox python=3.8
激活新环境
conda activate rppg-toolbox
安装pytorch
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
这行命令来自于:https://pytorch.org/get-started/previous-versions/ 中适配pytorch1.12.1
在命令行输入以下命令,可以看到,现在已经支持sm_86了
python -c "import torch; print(torch.__version__); print(torch.cuda.get_arch_list())"
显示
最后进入rPPG-Toolbox-main文件夹下
运行命令安装所需要的包即可。
pip install -r requirements.txt
即可完成环境安装。