人工智能学习之PyTorch环境快速安装

人工智能学习之PyTorch环境快速安装

前言

因为最近频繁的安装PyTorch环境，怕自己以后忘记，专门搞一个笔记来记录一下安装过程
主要是避免安装到cpu版本，要能使用CUDA的。

本文原创，转载请注明！！！

前提

默认已经安装好了 AnaConda，和Nvidia的显卡驱动+CUDA驱动

本次安装的PyTorch是官网最新版，可以使用CUDA

在cmd中输入 nvcc -V 查看cuda版本：

Conda

此部分用来说明conda的操作，转载自：https://zhuanlan.zhihu.com/p/44398592

常用指令

• 创建新环境

conda create --name pytorch python=3.11

• 查看所有环境

conda info -e
conda env list

• 激活/取消环境

activate pytorch
deactivate pytorch

• 删除环境

conda remove --name pytorch --all

• 其余常用指令

#查看已经安装的包
conda list
#安装包
conda install numpy
#安装 如果不用-n指定环境名称，则被安装在当前活跃环境
conda install -n pytorch numpy

PyTorch

首先用conda创建一个基础的环境并激活：

conda create --name pytorch python=3.11
activate pytorch

然后打开pytorch官网： https://pytorch.org/

在上边找到 Get Started，然后点击 Start Locally 或者 Previous PyTorch Versions 前者默认最新版，后者是历史版本，我个人喜欢使用特定版本 v2.0.1。

找到对应的指令，个人推荐使用pip安装，因为Linux下本人更常用pip。也可以提前下载好pip安装的whl包，然后直接本地使用pip安装。

这里就使用pip直接安装。注意这里要添加官网的下载地址，要不然pip默认安装是cpu版本的。

pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url  https://download.pytorch.org/whl/

碎碎念：

其实选择官网下载的原因是，因为有些版本，官网会使用链接安装，例如： --index-url https://download.pytorch.org/whl/ 链接，这个链接下载的版本一般都是可以直接使用CUDA的，而使用pip直接从镜像库里安装的一般都是只支持cpu，CUDA不可用。

安装完成之后，测试一下，在cmd已经激活的环境中，输入 python ，然后运行以下代码：

import torch
print(torch.__version__)
print(torch.cuda.is_available())

只要最后一个是True就可。