1、torch1.1.0(cuda9.0)
pip install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-linux_x86_64.whl
这是我最常用的版本。
2、torch1.6.0
pip install torch==1.6.0+cu92 torchvision==0.7.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
但是这个版本的pytorch在cu08上无法使用gpu。
3、tensorflow1.15
bert-as-service项目需要的版本
pip install --user tensorflow-gpu==1.15
4、tensorflow2.2.0
pip install tensorflow-gpu==2.2.0