pytorch 0.4.1+ubuntu18.04 +python3.67+cuda9.2 安装

版权声明:文章为网上公开资源结合个人情况修改整理。如有侵权,请联系删除。 https://blog.csdn.net/qq_41895190/article/details/83514752

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 


测试安装是否成功
 python命令下输入:

import torch

print(torch.__version__)

如果不报错,就是安装成功!

如果输出0.4.1,那么恭喜Windows下的PyTorch0.4.1安装成功!

最后需要安装 torchvision:

pip3 install torchvision


 

pytorch 0.4.1+win10 +python3.66+cuda9.2 安装

win10版:https://blog.csdn.net/qq_41895190/article/details/82761162 

猜你喜欢

转载自blog.csdn.net/qq_41895190/article/details/83514752