pip安装Python3.6下各版本pytorch命令

_我走路带风

于 2018-03-26 13:29:59 发布

阅读量2w

点赞数 4

CC 4.0 BY-SA版权

分类专栏：自然语言

本文链接：https://blog.youkuaiyun.com/qq_36652619/article/details/79696905

自然语言专栏收录该内容

61 篇文章

订阅专栏

本文提供了针对不同Python版本和CUDA配置下PyTorch及其依赖包torchvision的具体安装指令。覆盖了从Python3.6结合不同版本CUDA（无CUDA、CUDA8、CUDA9及CUDA9.1）的情况，通过直接指定URL进行安装。

Python3.6 + cuda none
pip3 install http://download.pytorch.org/whl/cpu/torch-0.3.1-cp36-cp36m-linux_x86_64.whl

pip3 install torchvision

Python3.6 + cuda 8
pip3 install http://download.pytorch.org/whl/cu80/torch-0.3.1-cp36-cp36m-linux_x86_64.whl

pip3 install torchvision

Python3.6 + cuda 9
pip3 install http://download.pytorch.org/whl/cu90/torch-0.3.1-cp36-cp36m-linux_x86_64.whl

pip3 install torchvision

Python3.6 + cuda 9.1
pip3 install http://download.pytorch.org/whl/cu91/torch-0.3.1-cp36-cp36m-linux_x86_64.whl

pip3 install torchvision