如果电脑有GPU,就安装 pytorch GPU 版本,可以加速
如果没有就安装CPU版本,执行速度可能会慢
首先,安装的pytorch版本与python版本有关系,对应关系如下:
其次,PyTorch 版本要根据 CUDA 版本决定,根据显示的CUDA,能安装什么版本的CUDA根据显卡决定
官网链接:https://pytorch.org/get-started/previous-versions/
PyTorch 版本 | CUDA 环境 |
---|---|
0.4.1、1.2.0、1.4.0、1.5.0(1)、1.6.0、1.7.0(1) | 9.2 |
1.2.0、1.1.0、1.0.0(1) | 10.0 |
1.4.0、1.5.0(1)、1.6.0、1.7.0(1) | 10.1 |
1.5.0(1)、1.6.0、1.7.0(1)、1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0、1.12.0(1) | 10.2 |
1.7.0(1) | 11.0 |
1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0 | 11.1 |
1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0、1.12.0(1) | 11.3 |
1.11.0 | 11.5 |
1.12.0(1) | 11.6 |
PyTorch下载地址:
https://download.pytorch.org/whl/torch_stable.html
选择对应版本,比如:cpu/torch-1.3.1%2Bcpu-cp37-cp37m-linux_x86_64.whl
然后cd到你的安装包目录下,使用
pip install XXX
如果安装失败,提示什么ssl在python中未开启,可能是国外网原因,使用如下命令
pip install XXX -i http://pypi.douban.com/simple --trusted-host pypi.douban.com