Docker容器安装Pytorch及第三方库

1. 容器内安装Pytorch

1.运行容器

1.|xiaoyuanzi@xiaoyuanzi-virtual-machine:~$ docker run -it new_python:3.6 /bin/bash  # 运行容器

2 .进入pytorch官网，选择使用pip下载。

2.|root@8edaa4ddac2a:/# pip3 install torch torchvision torchaudio -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

3 .验证是否下载成功

root@8edaa4ddac2a:/# python
Python 3.6.15 (default, Oct 13 2021, 09:43:57) 
[GCC 10.2.1 20210110] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import torchvision
>>> x = torch.rand(5,3)
>>> print(x)
tensor([[0.5893, 0.3121, 0.1701],
        [0.3873, 0.1000, 0.6725],
        [0.2716, 0.9126, 0.5763],
        [0.1469, 0.1019, 0.2020],
        [0.1365, 0.3138, 0.2949]])
>>> 

• 检查PyTorch GPU版本是否安装成功：

  >>> import torch
  >>> print(torch.__version__)
  0.4.0
  >>> print(torch.cuda.is_available())
  True
  

• 更新Pytorch：

   pip install pytorch==0.1.10
  

• 卸载Pytorch：

   pip uninstall torch
  

• 问题：

   no space left on device
  

• 解决1： 清理一下缓存

  1.|sudo apt-get autoclean
  2.|sudo apt-get clean
  

• 解决2： 更改缓存路径，如果你怀疑是缓存空间过小的原因

  1.|  sudo su
  2.|  export TMPDIR=/home/<username>/tmp
  3.|  ~/.bashrc
  

• 其他解决方法：

https://blog.csdn.net/jiedao_liyk/article/details/78497625?