说明:本教程针对有一定编程基础和项目基础的同学,需要提前配置好属于你自己的深度学习环境,在进行以下学习之前,请确保你的深度学习环境没问题。
1 安装PyTorch
1.1 安装Anaconda
Anaconda
是一个用于科学计算的
Python
发行版,支持
Linux, Mac, Windows,
包含了众多流行的科学计算、数据分析的 Python
包。
1.
去官方地址下载好对应的安装包
下载地址
:
https://www.anaconda.com/download/#linux
2.
安装
anaconda
bash ~/Downloads/Anaconda3-2021.05-Linux-x86_64.sh
anaconda
会自动将环境变量添加到
PATH
里面,如果后面你发现执行
conda
提示没有该命令,那么
你需要执行命令
source ~/.bashrc
更新环境变量,就可以正常使用了。
如果发现这样还是没用,那么需要添加环境变量。
编辑
~/.bashrc
文件,在最后面加上
export PATH=/home/bai/anaconda3/bin:$PATH
保存退出后执行:
source ~/.bashrc
再次输入
conda list
测试看看,应该没有问题。
1.2 添加Aanaconda国内镜像配置
清华
TUNA
提供了
Anaconda
仓库的镜像,运行以下命令
:
conda config --add channelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --add channelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/conda config --set show_channel_urls yes
1.3 安装pytorch
首先创建一个
anaconda
虚拟环境,环境名字可自己确定,这里本人使用
mypytorch
作为环境名
:
conda create -n mypytorch python=3.8
安装成功后激活
mypytorch
环境:
conda activate mypytorch
在所创建的虚拟环境下安装
,
执行命令:
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch