pytorch 0.4.1+ubuntu18.04 +python3.67+cuda9.2 安装
点击进入官网 https://pytorch.org/
方法一:pip安装
Python3.6+pip安装cpu版本
pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-linux_x86_64.whl pip3 install torchvision
Python3.6+pip安装gpu版本
目前gpu版本支持cuda8.0,cuda9.0和cuda9.2,请选择对应的版本下载安装,不要同时执行下面三个命令!
pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-linux_x86_64.whl pip3 install torchvision
(我选的是这个)
方法二:Conda安装
如果你是Anaconda|Python用户,cpu版本 执行命令:
conda install pytorch-cpu torchvision-cpu -c pytorch
就可以完成安装。
,如果你需要安装cuda8.0、cuda9.0或cuda9.2的gpu版本,请执行:
conda install pytorch torchvision cuda92 -c pytorch
pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-win_amd64.whl
测试安装是否