目录
安装PyTorch之前，强烈建议先在Ubuntu中安装Anaconda，从而更好的管理PyTorch需要的包及包对应的版本。安装过程可以参考：超详细Ubuntu安装Anaconda步骤+Anconda常用命令：https://blog.csdn.net/KRISNAT/article/details/124041869
STEP1：进入PyTorch官网查看安装版本和指令
进入PyTorch官网查看安装指令，不同的系统和环境对应的安装指令不同，官网会自动给出最佳的安装指令，如下图所示：
我这里官网给出的最佳安装指令如下：
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
复制这个安装指令备用。
STEP2：为PyTorch单独创建conda环境
使用STEP1中的指令安装PyTorch之前，需要先创建一个单独的conda环境，用于匹配对应的PyTorch版本。这一步不是必须的，但可以很方便的为PyTorch创建一个干净且独立的Python环境。这里使用比较稳定的3.6版本的Python解释器，创建代码如下：
conda create -n pytorch1110 python=3.6
我这里安装的是PyTorchan1.11.0稳定版，因此创建的conda环境命名pytorch1110，并指定了Python版本为3.6。安装结果如下，期间输入y确定安装一些必要的包。
STEP3：进入STEP2中创建的conda环境
输入以下命令即可进入对应环境：
conda activate pytorch1110
STEP4：输入STEP1中的安装指令安装PyTorch
在已经进入conda环境的终端中运行STEP1中的安装指令，如下：
因为网络原因，安装过程可能出错，如下：
出现以上问题，我们只需要再次运行刚刚的安装指令即可，安装成功结果如下：
至此，恭喜你已经完成了Ubuntu中PyTorch的安装，如果需要使用Jupyter Notebook，则需要完成STEP5。
STEP5：安装Jupyter Notebook需要的环境包
在刚刚安装PyTorch的conda环境中要使用Jupyter notebook或者Jupyter Lab需要单独安装，安装指令如下：
conda install nb_conda -y
运行完成指令就可以愉快的在交互式的Jypyter Notebook中测试自己的深度学习网络了。
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 举报，一经查实，本站将立刻删除。
文章由极客之音整理，本文链接：https://www.bmabk.com/index.php/post/74508.html
