目录
- 一、anaconda 安装
- 二、创建pytorch
- 1. 创建pytorch环境:
- 2. 激活环境
- 3. 下载安装pytorch包
- 4. 检查是否安装成功
一、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()
效果如下图所示既表示安装成功!
先写到这里吧,到这里就基本安装成功了,如果有什么报错可以自行百度解决。
最后祝大家都可以顺利安装成功,成为一名优秀的算法工程师!