我安装pytorch用了一个下午和晚上。。。。

安装过程:

目录

安装anaconda

安装pytorch

查看电脑是否支持CUDA:

支持CUDA的电脑:(不支持的自动略过)

anaconda安装pytorch

 创建虚拟环境:

切换到该环境

输入安装命令,命令是在pytorch官网中生成的

测试安装成功:


安装anaconda

网址:Anaconda | Individual Edition

进入之后,下滑到如图,当前有显示合适的python版本,

如果你的和你的python版本不同,则点击下图的 archive,会有一系列的版本适配你的python

李沐anaconda cuda pytorch d2l 安装 anaconda怎么装pytorch_python

李沐anaconda cuda pytorch d2l 安装 anaconda怎么装pytorch_CUDA_02

然后开始下载安装,安装过程可以看这篇博客 链接:anaconda安装配置教程_振华OPPO的博客世界-_anaconda安装配置

安装pytorch

 去官网下载:本地启动|PyTorch

李沐anaconda cuda pytorch d2l 安装 anaconda怎么装pytorch_CUDA_03

下滑,找到这个图像,每一行的意思如下:

产品的稳定性:选择产品的稳定性,一般选stable(稳定)

系统:

下载方式:较多的是Conda(需要下载anaconda),pip(一般python就自带)

语言:看你用的语言

CUDA/CPU:如何选择,这要看电脑显卡配置

查看电脑是否支持CUDA:

此电脑右击管理->设备管理器->显示适配器 

李沐anaconda cuda pytorch d2l 安装 anaconda怎么装pytorch_pytorch_04

 我的电脑就不支持CUDA,所以我选择的就是CPU。

支持CUDA的电脑:(不支持的自动略过)

支持CUDA的显示适配器一般是有两行的,

李沐anaconda cuda pytorch d2l 安装 anaconda怎么装pytorch_pytorch_05

 那如何查看CUDA的版本:打开CMD,输入nvcc --version

anaconda安装pytorch

在开始找到如下 双击打开:

李沐anaconda cuda pytorch d2l 安装 anaconda怎么装pytorch_官网_06

 创建虚拟环境:

创建名为pytorch,python版本为3.9的环境,环境名字可以自己设置

conda create -n pytorch python=3.9

切换到该环境

conda activate pytorch

李沐anaconda cuda pytorch d2l 安装 anaconda怎么装pytorch_pytorch_07

  

输入安装命令,命令是在pytorch官网中生成的

conda install pytorch torchvision torchaudio cpuonly -c pytorch

回车后,会需要你选择y/n,选择y,此时开始下载 

李沐anaconda cuda pytorch d2l 安装 anaconda怎么装pytorch_pytorch_08

当屏幕上显示done,则安装完成 

 

李沐anaconda cuda pytorch d2l 安装 anaconda怎么装pytorch_anaconda_09

速度可能会比较慢,可以考虑切换源,切换后,安装命令的 -c pytorch则删去

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
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/cloud/msys2/
 
# 设置搜索时显示通道地址
conda config --set show_channel_urls yes

切回默认源: 

conda config --remove-key channels

测试安装成功:

pip list

李沐anaconda cuda pytorch d2l 安装 anaconda怎么装pytorch_官网_10

在刚刚创建的pytorch环境命令行中输入python 进入python,没有反应会是没有显示则成功,然后再输入torch.cuda.is_available(),  安装的是CPU版本的则显示false,CUDA的则显示true

李沐anaconda cuda pytorch d2l 安装 anaconda怎么装pytorch_pytorch_11