ubuntu20.04在miniconda中安装pytorch1.7

ubuntu20.04+cuda11.0+cudnn8.0.5+pytorch1.7.1+python3.8
1.安装miniconda;
2.更改miniconda源
参考链接: 在Ubuntu的miniconda中如何加入国内镜像源.
3.创建虚拟环境
参考链接:Pytorch1.7.0—GPU安装教程.

conda create -n pytorch_1.7 python=3.8  #创建虚拟环境
activate python_1.7  #激活进入虚拟环境

4.在虚拟环境中安装pytorch

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0  #安装pytorch,注意后缀“-c pytorch”去掉
conda install ipykernel  #安装ipykernel库,用来注册
python -m ipykernel install --user --name pytorch_1.7 --display-name pytorch_1.7  #将环境名注册到notebook
conda install notebook  #安装notebook

你可能感兴趣的:(pytorch,python,深度学习)