0


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 

本文转载自: https://blog.csdn.net/weixin_43717144/article/details/124822798
版权归原作者 HappyLaber 所有, 如有侵权,请联系我们删除。

“Python-安装Pytorch”的评论:

还没有评论