- 首先安装cuda
- 安装cudnn
- 安装pytorch,链接如下:
pip install -i http://pypi.douban.com/simple torch==1.5.1 torchvision==0.6.1 -f https://download.pytorch.org/whl/torch_stable.html --trusted-host pypi.douban.com
如果需要安装不同版本,请到官网查看版本号,再修改上述链接版本即可。
官网链接:https://pytorch.org/get-started/locally/