在Anaconda Jupyter Notebook中如何安装Pytorch

Source


pytorch环境

1.进入Anaconda Prompt以后默认是base环境
在这里插入图片描述
2.如何创建环境
conda是指调用conda包,create是指创建屋子,-n是指后面的名字是屋子的名字,pytorch是指屋子的名字,python=3.8是指创建的屋子是python3.8版本
如何查看Anaconda中的python版本?
进入Anaconda Prompt以后输入python -V
在这里插入图片描述
首先使用conda指令创建一个屋子
conda create -n pytorch python=3.8
之后弹出提示输入y即可安装
3.如何查看环境
安装成功以后输入
conda info --envs
在这里插入图片描述
可以看到conda环境中有新建的pytorch环境
在pytorch环境中安装PyTorch,使用如下指令进入pytorch环境
conda activate pytorch
在这里插入图片描述
conda activate name可以进入环境
conda activate可以退出当前环境
conda remove -n name --all可以删除环境

pytorch安装

pytorch官网:https://pytorch.org/
PyTorch Build选择Stable(1.8.1);Your OS选择Windows;Package选择Conda;Language选择Python;Computer Platform选择CYDA10.2
Package Windows下推荐Conda,Linux下推荐Pip
在这里插入图片描述
打开Anaconda Prompt进入pytorch环境,复制Run this Command中的命令conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch粘贴到pytorch环境中
在这里插入图片描述
之后弹出提示,输入y即可安装,pytorch安装较慢,等着它慢慢下。如果安装过程中出现网络中断或者安装不上的情况下,重新复制PyTorch安装Run this Command中的命令,粘贴到pytorch环境中,重新进行安装(我就重新安装了好几次)

验证是否安装成功

1.打开Anaconda Prompt进入pytorch环境
2.输入python
3.输入import torch如果没有报错就意味着PyTorch已经顺利安装
4.输入==torch.cuda.is_available()==如果是True就意味着可以使用GPU;如果是False就意味着只能使用CPU
在这里插入图片描述

torch.cuda.is_available() 返回是False如何解决

1.更新显卡驱动至当前最新版本
官网:https://www.nvidia.cn/geforce/drivers/
在这里插入图片描述
2.CUDA Toolkit与Driver Versions版本不同
我将显卡驱动更新至当前最新版本以后还是报错,我查看了好多资料,有的博主说可能是版本不同
在这里插入图片描述

我安装的torch版本是1.8.1,CUDA10.2要求>=440.33,但是我的只有425.31,所以我只能下载CUDA10.1
在这里插入图片描述
pytorch版本下载链接:
https://pytorch.org/get-started/previous-versions/
我卸载了1.8.1,并下载了1.7.1中的CUDA10.1
在这里插入图片描述
卸载pytorchpip uninstall torch
重新安装pytorch复制CUDA10.1下面的命令粘贴到pytorch环境中
在这里插入图片描述
在这里插入图片描述
torch.cuda.is_available() 返回是True
在这里插入图片描述

Jupyter Notebook调用pytorch虚拟环境

1.以管理员身份打开Anaconda Prompt
在这里插入图片描述
2.输入conda install nb_conda之后弹出提示,输入y即可安装
在这里插入图片描述

3.进入pytorch环境输入conda install ipykernel之后弹出提示,输入y即可安装
在这里插入图片描述
4.打开Jupyter Notebook会发现增加了两项
在这里插入图片描述
5.点击Kernel-Change kernel-Python[conda env:pytorch]
在这里插入图片描述
在这里插入图片描述

6.右上角显示Python[conda env:pytorch]即可
在这里插入图片描述
认真是一种态度更是一种责任