一、 环境配置
之前配置的cuda11.3很多包版本不一致导致程序无法正常运行,又重新安装了cuda9.0,并更换了python版本,但是折腾半天显示显卡不可用,最后查找各种资料才知道30系列显卡不支持11.0以下的cuda,即使安装成功也不能调用,因此最终还是使用了CUDA11.3配置环境,python选3.6,pytorch版本1.10.0
conda create --name DF-VO113 python=3.6
conda activate DF-VO113
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge
【以下部分为参考内容,与DF-VO环境配置无关,可跳过 】
=========================================================================
安装pytorch,查找对应cuda的版本
复制对应命令并下载,但conda报错