pytorch

 运行PyTorch的安装命令

如果你和我一样，采用8.0CUDA，2.7版本Python，可以运行以下命令：

conda install pytorch torchvision cuda80 -c soumith

如果没有显卡，2.7版本Python，可以这样：

conda install pytorch torchvision -c soumith

 

5. 验证安装是否成功

要显示显卡信息，在终端输入：

sudo lshw -c video

 

要显示CUDA信息，在终端输入：

nvcc -V

 

查看Phthon版本，在终端输入：

python --version

 

验证pytorch是否安装成功，在终端输入：

python

此时进入python环境。然后，继续输入

import torch

import torchvision

不报错就表明安装成功。
--------------------- 
 

 

××××××××××××××××××××××××××××××××××××××××××××××××××××××

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
--------------------- 
作者：_我走路带风 
来源：优快云 
原文：https://blog.youkuaiyun.com/qq_36652619/article/details/79696905 
版权声明：本文为博主原创文章，转载请附上博文链接！