2024小白安装Pytorch-GPU版(Anaconda,CUDA,cuDNN讲解)

目录

前言

一、正式安装Pytorch前的准备

二、安装Anaconda

1.下载

2.安装注意

3.安装成功

4.搭建虚拟环境 

三、安装CUDA(pytorch-GPU版本)

1.查看CUDA显卡版本

2.下载

3.安装步骤(只介绍CUDA官网下载)

四、安装cuDNN

1.下载

2.安装步骤

五、安装Pytorch

1.下载

2.安装步骤

1.复制Pytorch安装指令

2.进入需要安装Pytorch的虚拟环境

3.安装成功

六、遇到的问题

1.CUDA安装问题(nsight visual studio edition失败)

2. SSL证书问题

总结


前言

本帖子记录自己作为一个0基础小白安装pytorch的过程,和遇到的一些问题及解决办法。

系统:Windows,有NVDI显卡


一、正式安装Pytorch前的准备

想要安装Pytorch我们需要在这之前安装好Anaconda,CUDA和cuDNN,那为什么要安装这些呢?

Pytorch介绍

PyTorch是一个基于Python的深度学习框架,用于构建和训练神经网络模型

Anaconda介绍

Anaconda提供了一个便捷的环境管理器conda,可以用于在同一个机器上安装不同版本的软件包(python)及其依赖,并能够在不同的环境之间切换。

CUDA介绍

CUDA是NVIDIA推出的并行计算平台和API模型,它使得显卡可以用于图像渲染和计算以外的目的。PyTorch通过CUDA可以充分利用GPU的计算能力,加速深度神经网络的学习和推理过程。

cuDNN介绍
cuDNN是CUDA的扩展库,专门针对深度神经网络中的基础操作提供高度优化的实现方式,例如卷积、池化、规范化以及激活层的前向和后向过程。使用cuDNN可以大大提高深度学习模型在GPU上的运行效率。

通过上面介绍我们知道要用Pytorch就需要三个工具帮忙,一个用来管理要用的环境(Anaconda),两个用来加速计算(CUDA,cuDNN是建立在CUDA上的)


二、安装Anaconda

1.下载

官方地址(最新版):Download Now | Anaconda

历史版本地址:Download Now | Anaconda

 下载Anaconda会默认安装一个与它对应的python版本,高版本的Anaconda兼容的低版本python版本。

2.安装注意

其他步骤就是正常next,然后选择装在哪个盘了,这里就不介绍了。

3.安装成功

这样就是安装成功了!

4.搭建虚拟环境 

 pytorch是安装在你需要的虚拟环境里,所以你可以创建一个新的虚拟环境,或者直接安装在默认的base中。楼主这里创建了一个python=3.8版本的环境

(1)创建新虚拟环境

conda create -n  自己取的环境名字  python=版本号

(2)删除虚拟环境

conda remove --name 环境名字 --all

(3)激活进入环境

conda activate 环境名字

(4)查看已存在的虚拟环境

conda env list


三、安装CUDA(pytorch-GPU版本)

1.查看CUDA显卡版本

在cmd命令行终端输入nvidia-smi

查看电脑是否有显卡其他方法:

2.下载

注意:下载的CUDA版本必须小于最高支持的版本号

CUDA官方地址:CUDA Toolkit Archive | NVIDIA Developer

3.安装步骤(只介绍CUDA官网下载)

 注意:这个临时路径和后面的安装路径不要一致,否则安装完会找不到CUDA文件

 默认选第一个就行,如果安装失败了,就要选第二个了。

 查看是否安装成功

打开cmd输入:nvcc -V


四、安装cuDNN

1.下载

 注意:CUDA的版本要与cuDNN版本能对应(楼主下的CUDA11.1,cuDNN下载对应11.x就行)

cuDNN官方地址:cuDNN Archive | NVIDIA Developer

注意:因为需要注册才能安装,我们选择复制链接到迅雷下载,跳过这个注册过程。

2.安装步骤

解压完,复制三个文件到CUDA文件下即可


五、安装Pytorch

1.下载

官方地址:PyTorch

历史版本:Previous PyTorch Versions | PyTorch

2.安装步骤

1.复制Pytorch安装指令

选conda和pip下载都行(两则只是管理不同),此处楼主选择的pip下载

 下面这个是楼主装的版本v1.8的Pytorch的pip指令

2.进入需要安装Pytorch的虚拟环境

 输入复制的指令即可进行安装

3.安装成功

输入:pip list


六、遇到的问题

1.CUDA安装问题(nsight visual studio edition失败)

解决参考链接:安装CUDA失败的情况nsight visual studio edition失败_cuda安装失败-优快云博客 

2. SSL证书问题

复制的Pytorch下载指令安装时出现如下:

Could not fetch URL https://pypi.tuna.tsinghua.edu.cn/simple/torch/: There was a problem confirming the ssl certificate: HTTPSConnectionPool(host='pypi.tuna.tsinghua.edu.cn', port=443): Max retries exceeded with url: /simple/torch/ (Caused by SSLError(SSLZeroReturnError(6, 'TLS/SSL connection has been closed (EOF) (_ssl.c:1149)'))) - skipping ERROR: Could not find a version that satisfies the requirement torch==1.8.0+cu111 (from versions: none)

 解决办法:

1.更换 PyPI 镜像源

2.更新 pip:在虚拟环境中输入pip install --upgrade pip

3.关闭防火墙

4.网络连接问题(关掉梯子)

总结

大家遇到问题可以csdn上看看其他大佬弄的或者问问AI,有时候也挺好用。祝大家学习愉快!

安装 PyTorch GPU 本需要先安装 CUDAcuDNN。以下是在 Ubuntu 20.04 上安装 PyTorch GPU 本的步骤: 1. 安装 CUDA 11.7 可以从 NVIDIA 官网下载 CUDA Toolkit 11.7 的安装包,选择适合自己系统的本进行下载,下载地址为:https://developer.nvidia.com/cuda-downloads 下载完成后执行以下命令进行安装: ``` sudo sh cuda_11.7.0_<version>_linux.run ``` 安装过程中需要选择安装路径,一般选择默认路径即可。 2. 安装 cuDNN 8.2.2 前往 NVIDIA 官网下载 cuDNN 8.2.2 的安装包,下载地址为:https://developer.nvidia.com/cudnn-download-survey 注册一个 NVIDIA 开发者账号后就可以下载了。 下载完成后解压文件,并将 cuDNN 目录下的文件复制到 CUDA安装目录下的对应文件夹中。 ``` cd <path_to_cudnn_folder> sudo cp -P include/cudnn*.h /usr/local/cuda-11.7/include sudo cp -P lib64/libcudnn* /usr/local/cuda-11.7/lib64/ sudo chmod a+r /usr/local/cuda-11.7/include/cudnn*.h /usr/local/cuda-11.7/lib64/libcudnn* ``` 3. 安装 PyTorch 推荐使用 Anaconda 或 Miniconda 进行安装。 先创建一个新的环境: ``` conda create --name pytorch_gpu python=3.9 conda activate pytorch_gpu ``` 然后安装 PyTorch: ``` conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch -c conda-forge ``` 安装完成后可以通过以下命令测试是否安装成功: ``` python -c "import torch; print(torch.cuda.is_available())" ``` 如果输出为 True,则说明安装成功。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值