【深度学习基础】环境搭建 linux系统下安装pytorch

一、anaconda 安装

具体的安装说明可以参考我的另外一篇文章【环境搭建】Linux报错bash: conda: command not found…

二、创建pytorch

1. 创建pytorch环境：

conda create -n dtc python=3.8

这里：

create 代表创建环境。
-n代表name， 也就是说我们创建的这个环境名字叫做dtc。可以根据自己的意愿设置环境名字。
python=3.8， 这里我们是指定python3.8版本，如果不指定版本的话会默认安装最新版的python，同时也可以根据自己的需求设置python版本。

此时我们名字为dtc的环境只是一个躯壳，下面我们要进入这个环境中并安装pytorch包。

2. 激活环境

conda activate dtc

可以看见，激活环境之后，我们由原先的base环境进入到被激活的dtc环境。

3. 下载安装pytorch包

然后去pytorch官网找到适合自己电脑配置的pytorch版本进行安装，在这里由于我是安装在服务器终端，所以选择了以下版本:

安装命令在官网上选好配置后会显示出来

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

将上述代码直接复制到终端即可开始安装pytorch包。

4. 检查是否安装成功

在pytorch环境下依次输入下列语句：

python #检查python版本
import torch
torch.cuda.is_available()
1
2
3

效果如下图所示既表示安装成功！

先写到这里吧，到这里就基本安装成功了，如果有什么报错可以自行百度解决。
最后祝大家都可以顺利安装成功，成为一名优秀的算法工程师！