ubuntu18.04从0开始安装pytorch1.6

ubuntu18.04安装pytorch1.6

---

前言

提示：这里可以添加本文要记录的大概内容：
例如：随着人工智能的不断发展，机器学习这门技术也越来越重要，很多人都开启了学习机器学习，本文就介绍了机器学习的基础内容。

---

提示：以下是本篇文章正文内容，下面案例可供参考

一、安装Anaconda

官网地址：
https://www.anaconda.com/products/individual
进入官网，点击Download按钮，跳转到底部，界面如下：
选择linux版本下的64-Bit (x86) Installer (544 MB)进行下载，下载完成后
打开终端输入以下命令：

bash Anaconda3-2021.05-Linux-x86_64.sh

一直按回车，让输入yes/no的时候输入yes即可，完成安装。

二、安装cuda与cudnn

1.配置显卡驱动

参考链接：

https://blog.csdn.net/qq_39407949/article/details/107841465
https://blog.csdn.net/qq_20492405/article/details/79034430

2.下载并安装

参考链接1：

参考链接：https://www.cnblogs.com/answerThe/p/12244354.html

cuda官网下载链接：
https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=18.04&target_type=runfile_local

下载格式自行选择，下载runfile(local)格式的文件的话运行的命令会少一些，偏向于Windows下图形化安装风格，参考链接用的也是这种方法；选择好格式后网页下方会出现下载、安装命令，如下所示，完成下面命令后的其他步骤请看参考链接。

wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
sudo sh cuda_10.2.89_440.33.01_linux.run

按照参考链接操作成功查到了cuda的版本号：
PS：我的驱动版本号高于cuda版本号，查了一下有博客说这种情况不影响，安装cuda的时候没有选择装驱动（详见参考链接1）：

参考链接2：

https://www.jianshu.com/p/eb5335708f2a

执行参考链接1中step5：在终端查看CUDNN版本 时出现异常，没有出现相应的信息，没有任何输出：
找到一个解决办法，但是并不适用，文件夹中并没有找到cudnn_version.h文件：
参考链接3：

成功解决–查看cudnn版本无反应的问题

找到了cuDNN官方安装文件，非常详细：
https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#prerequisites

官方网站对于cudnn安装成功的验证提供了另一种方法：
打开一个新的终端，输入以下命令：

cp -r /usr/src/cudnn_samples_v8/ $HOME
cd  $HOME/cudnn_samples_v8/mnistCUDNN
make clean && make
./mnistCUDNN

显示结果表明cudnn安装成功：

三、安装pytorch

pytorch所有历史版本官方安装链接：
https://pytorch.org/get-started/previous-versions/
cuda10.2安装pytorch1.6的命令如下：

# CUDA 10.2
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch

安装需要的时间比较久。
cuda安装的官网教程：
https://pytorch.org/get-started/locally/
按照官网进行测试遇到报错：

import-im6.q16: not authorized `torch' @ error/constitute.c/WriteImage/1037.

找到了错误的原因是没有启动python，启动后测试成功！
参考链接：

https://blog.csdn.net/qq_41874879/article/details/106122293

总结

到此为止，成功安装了anaconda、cuda、cudnn、porch!