目录
前言
本帖子记录自己作为一个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版本必须小于最高支持的版本号
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
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 pip3.关闭防火墙
4.网络连接问题(关掉梯子)
总结
大家遇到问题可以csdn上看看其他大佬弄的或者问问AI,有时候也挺好用。祝大家学习愉快!