pip安装pytorch cuda版本-清华源,可根据版本要求进行版本的替换
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.8.0+cu102 torchvision===0.9.0+cu102 -f https://download.pytorch.org/whl/torch_stable.html
pip安装pytorch cuda版本-清华源,可根据版本要求进行版本的替换
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.8.0+cu102 torchvision===0.9.0+cu102 -f https://download.pytorch.org/whl/torch_stable.html