深度学习PyTorch环境安装

一、Anaconda3

1.下载清华源的Anaconda3

https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/?C=M&O=D

2终端运行

bash Anaconda3....（下载的文件名）,一路enter

直到出现initialize输入yes

2.配置环境变量

which anaconda #找到安装好的conda路径
vim ~/.bashrc
export PATH="/conda的路径:$PATH"  #例如export PATH="/home/fyq/anaconda3/bin:$PATH"
source ~/.bashrc

3.添加源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

4.创建虚拟环境(pytorch为虚拟环境名字)

conda create -n pytorch python=3.9

5.进入虚拟环境

conda activate pytorch 

退出命令：conda deactivate pytorch

二、创建pytorch(可以在官网选择需要的版本生成对应命令),此命令为pytorch1.11,

!!!!!!!!!!注意，去掉命令行后面的-c pytorch，不然非常慢！！！

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

检验

进入虚拟环境后

python
import torch
torch.cuda.is_available()