Python-安装Pytorch

文章目录

注：本文参考了
深入浅出PyTorch

一.安装Anaconda

安装步骤略.

二.创建虚拟环境

1.打开Anaconda Prompt

2.创建虚拟环境

conda create -n env_name python==version
# env_name是虚拟环境的名字,我改为了deeplearning# version是python的版本,我选择的version为3.7.9

3.激活虚拟环境

conda activate deeplearning

4.更新pip和conda

pip3 install --upgrade pip
conda update conda

5.退出虚拟环境

conda deactivate

扩展知识：

1.查看现存虚拟环境

conda env list

2.删除某个虚拟环境

conda remove -n env_name --all

三.安装pytorch

1.查看是否有GPU

2.查看CUDA版本

3.获取Pytorch安装命令：PyTorch

4.打开Anaconda Promp

5.激活虚拟环境

conda activate deeplearning

6.在Anaconda Prompt中输入安装命令（安装命令见第3步）

% 以网页为准！这里只是个例子！
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

拓展：使用临时清华源

% 安装仅支持CPU的版本！
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch

7.验证Pytorch是否支持CUDA

python
import torch
torch.cuda.is_available()

8.退出虚拟环境

conda deactivate

四.配置jupyter notebook

1.打开Anaconda Promp

2.激活虚拟环境

conda activate deeplearning

3.为jupyter notebook加入虚拟环境内核

conda install nb_conda
conda install ipykernel
conda install -n deeplearning ipykernel
python -m ipykernel install --user --name deeplearning

4.启动jupyter notebook

jupyter notebook

拓展：添加插件

pip install jupyter_contrib_nbextensions -i https://pypi.tuna.tsinghua.edu.cn/simple
jupyter contrib nbextension install --user
pip install jupyter_nbextensions_configurator -i