linux上部署torch(CPU/GPU)

╰+攻爆jí腚メ 2021-11-10 06:14 878阅读 0赞

linux上部署的pytorch和python的版本和cuda的版本有关系。因此在安装pytorch需要根据自己系统上的部署的cuda和python版本来部署pytorch。

## 注意：使用`pip`或`pip3`安装时要注意其对应的是哪个版本的`python` ##

查看方式为：pip -V

## 一、部署pytorch(CPU/GPU) ##

**1、部署pytorch（GPU）版**  
pytor的官网如下  
[https://pytorch.org/get-started/locally/][https_pytorch.org_get-started_locally]  
在官网中根据自己的环境安装pytorch，因为我系统上没有安装Anaconda因此需安装来源选择使用pip安装。![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM5ODUyNjc2_size_16_color_FFFFFF_t_70]  
如上图所示，系统自带python2和python3版本，在python3上安装时需要安装python-pip3

sudo apt-get install python3-pip

注意：安装过程中要按住两个文件。

pip3 install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp35-cp35m-linux_x86_64.whl
    pip3 install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp35-cp35m-linux_x86_64.whl

**2、部署pytorch的CPU版**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM5ODUyNjc2_size_16_color_FFFFFF_t_70 1]  
注意：安装过程中要按住两个文件。

pip3 install https://download.pytorch.org/whl/cpu/torch-1.1.0-cp36-cp36m-linux_x86_64.whl
    pip3 install https://download.pytorch.org/whl/cpu/torchvision-0.3.0-cp36-cp36m-linux_x86_64.whl

## 二、查看python和CUDA版本 ##

**安装完成之后重新检查一下是否安装在python3环境下，以及是否和cuda版本想对应。**  
查看python

python\python3  或 python -V |  python3 -V

![在这里插入图片描述][20190802143133691.png]  
![在这里插入图片描述][20190802143153385.png]  
查看cuda版本

nvcc -V

![在这里插入图片描述][20190802143057418.png]

## 三、验证是否安装pytorch完成 ##

python3
    import pytorch
    import torchvison

![在这里插入图片描述][20190802143219991.png]  
发现在`import torchvision`时出现以下错误：

Detected that PyTorch and torchvision were compiled with different CUDA versions. PyTorch has CUDA Version=9.0 and torchvision has CUDA Version=10.0. Please reinstall the torchvision that matches your PyTorch install.

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM5ODUyNjc2_size_16_color_FFFFFF_t_70 2]

## 解决办法： ##

pip3 install --no-deps torchvision

重新验证`torchvision`：  
![在这里插入图片描述][20190802143746228.png]  
完美解决。  
最后，验证GPU是否可用  
![在这里插入图片描述][20190802143251282.png]  
返回True，表明已经可以使用GPU了！  
![在这里插入图片描述][20190802144407180.png]

[https_pytorch.org_get-started_locally]: https://pytorch.org/get-started/locally/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM5ODUyNjc2_size_16_color_FFFFFF_t_70]: /images/20211109/85ac38e127364de1aa48d811e44e257f.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM5ODUyNjc2_size_16_color_FFFFFF_t_70 1]: /images/20211109/4a2027c22e924e7485eb5f5e1276f05e.png
[20190802143133691.png]: /images/20211109/b6ef4bb26d054a46acb1961f80f379aa.png
[20190802143153385.png]: /images/20211109/5aa0344dbdf044d083197b5992d34fb6.png
[20190802143057418.png]: /images/20211109/10f1cefabaf448d29b8009c21899b3c9.png
[20190802143219991.png]: /images/20211109/60eeb93b6a794306b27278ed10a04385.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM5ODUyNjc2_size_16_color_FFFFFF_t_70 2]: /images/20211109/4e13f2b2b95944958b240cfd7bb97900.png
[20190802143746228.png]: /images/20211109/d4ee073847e54c15862324e27ed1170f.png
[20190802143251282.png]: /images/20211109/fbf7bd0cfc6b4dc28d736d2339e7f1da.png
[20190802144407180.png]: /images/20211109/70d6af9e0e4f403c947b8f1acbe5c3a7.png