建立PyTorch虚拟环境

以FairMOT为例:python conda create -n FairMOT python=3.7 conda activate FairMOT conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch (这一步指定pytorch版本为1.2.0,torchvision0.4.0,CUDA10.0)
相关文章
相关标签/搜索