记录一次简单、高效、无错误的linux上安装pytorch的过程

刺骨的言语ヽ痛彻心扉 2023-06-18 04:52 11阅读 0赞

# 1 准备miniconda #

Miniconda  
Miniconda 可以理解成Anaconda的免费、浓缩版。它非常小，只包含了conda、python以及它们依赖的一些包。我们可以根据我们的需要再安装相关的包，非常推荐。

## 1.1 下载 ##

官方地址在这里：  
[https://docs.conda.io/en/latest/miniconda.html][https_docs.conda.io_en_latest_miniconda.html]

如果下载linux下的Python 3.7的安装包，可以直接点击：  
[https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86\_64.sh][https_repo.anaconda.com_miniconda_Miniconda3-latest-Linux-x86_64.sh]

你在64位的linux下可以直接使用如下命令获取安装包：

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh .

## 1.2 安装 ##

然后安装只需要运行下面的命令，再根据提示傻瓜式安装即可：

sh Miniconda3-latest-Linux-x86_64.sh

## 1.3 配置 ##

安装好了之后，你需要退出当前的命令行，再重新打开新的命令行，这是你就会发现你的命令行会是这样形式的：

（base）[XXXX@XXX]

这里的base是miniconda默认创建的虚拟环境，如果你想退出的话，可以使用下列的命令退出当前的虚拟环境

conda deactivate

### 1.3.1 取消默认base虚拟环境 ###

现在只要你一打开命令行就会默认进入`base`虚拟环境，有的人非常不喜欢这样，比如我。有什么办法可以取消默认进入虚拟环境么？有的。只要通过使用下列命令后，退出当前命令行，再进入命令行，就会发现，已经不会默认进入`base`虚拟环境了。

conda config --set auto_activate_base false

### 1.3.2 设置conda镜像源 ###

通常conda默认的镜像源，下载速度缓慢，所以我们需要修改一下conda默认的镜像源，可以使用vim编辑`~/.condarc`，将里面的内容修改成下面的内容，然后保存退出（这块内容，我在之前的[博文][Link 1]也有讲过）：

channels:
      - defaults
    show_channel_urls: true
    default_channels:
      - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
      - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
      - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
    custom_channels:
      conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

# 2 安装pytorch #

## 2.1 创建pytorch虚拟环境 ##

我们可以通过下列命令查看当前用户下存在多少的conda虚拟环境：

conda info -e

然后我们通过如下命令创建`pytorch`的虚拟环境

conda create -n pytorch

## 2.2 安装pytorch ##

安装`pytorch`需要你知道自己的服务器基本信息，比如有GPU么？cuda的版本是多少？然后再相应的采用不同的命令进行安装，这里可以参考官方网站给出的安装命令：[https://pytorch.org/get-started/locally/][https_pytorch.org_get-started_locally]

想看你的电脑的CUDA版本，可以采用如下命令：

cat /usr/local/cuda/version.txt

我的服务器是有GPU的，cuda是9.2的，python版本是刚才通miniconda安装的python3.7，所以我是这样选择的：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9zdHk5NDUuYmxvZy5jc2RuLm5ldA_size_16_color_FFFFFF_t_70]  
这里给出的命令是：

conda install pytorch torchvision cudatoolkit=9.2 -c pytorch

**但是我使用这条命令安装总是出错，后来我改成下列的命令完美安装成功了：**

conda install pytorch torchvision cudatoolkit=9.2

**是的，我去掉了`-c pytorch`**

为什么这样就行了呢？按照这位博主的[解释][Link 2]**因为使用`-c pytorch`命令pytorch的安装源并没有使用我们指定的清华的安装源**

# 最后祝你深度学习愉快 #

## 觉得本文对你有帮助，可以在文章右侧点赞支持一下，也欢迎在网页左上方处点击关注，感谢支持 ##

[https_docs.conda.io_en_latest_miniconda.html]: https://docs.conda.io/en/latest/miniconda.html
[https_repo.anaconda.com_miniconda_Miniconda3-latest-Linux-x86_64.sh]: https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
[Link 1]: https://blog.csdn.net/sty945/article/details/103314201
[https_pytorch.org_get-started_locally]: https://pytorch.org/get-started/locally/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9zdHk5NDUuYmxvZy5jc2RuLm5ldA_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20191130111340640.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9zdHk5NDUuYmxvZy5jc2RuLm5ldA==,size_16,color_FFFFFF,t_70
[Link 2]: https://blog.csdn.net/jiyangsb/article/details/82430794