pytorch和tensorflow几种版本的下载链接

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
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。