Pytorch-Encoding安装

  • 较新版本的pytorch不支持
  • 要求python>=3.8
conda create -n torch1.4 python=3.8
source activate torch1.4
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch
cd PyTorch-Encoding
python setup.py install
  • Pycharm运行中可能需要做如下修改(多处):
    为了解决cannot import name ‘cpu’ from partially initialized module ‘encoding’ (most likely due to a circular import)
    Pytorch-Encoding安装_第1张图片

你可能感兴趣的:(环境,pytorch)