Pytorch最新安装教程(anaconda\python\CUDA\pytorch)
目录
一、写在前面
二、Anaconda/Python安装配置
三、CUDA配置准备
四、Pytorch安装
五、测试验证
一、写在前面
如果你遇到下面几个问题,那么这篇教程就非常适合你:
- 搜到的已有教程太多,难以分辨
- cuda/cudatoolkit/cudnn和pytorch/torchaudio/torchvision对应版本不会选
- pytorch最新更新的版本里没有conda install链接,搜到的教程都没有提到这一点
- pytorch安装太慢,镜像源也不行
以上教程是基于已有教程和自己经验的总结,分为四个部分,如果帮助请多多点赞和转发。
二、Anaconda/Python安装配置
这一步主要安装VScode/Anaconda/Python三个,链接如下:
下载完成之后,我参考的是如下的安装教程,十分详细,按步骤照做就行。
全网最简约的Anaconda+Python3.7安装教程Win10(百分百成功)_anaconda python3.7-CSDN博客
三、CUDA配置准备
这一步是CUDA的相关配置,主要是cudatoolkit和cudnn的安装,这一步关键是CUDA的版本对应(千万不能下载高于你电脑适配版本的CUDA),详细教程参考如下:
https://www.python.org/downloads/
四、Pytorch安装
这一步是pytorch/torchaudio/torchvision的适配安装,但搜索已有的教程,大部分都是conda install之类,但是目前最新pytorch版本失效;采用pip install可能会安装到C盘,且时间很长可能失败。
我采用的是下载离线适配版本+创建虚拟环境安装,下载很快且很实用。
1.寻找适配版本,可以参考以下链接里的表格(根据python和cuda的版本)
深度学习 | pytorch + torchvision + python 版本对应及环境安装_pytorch对应的python版本-CSDN博客
2.创建虚拟环境,下载和安装这三个包,下载地址:download.pytorch.org/whl/cu118
具体细节可以参考PyTorch离线版本安装(超详细版!)_pytorch离线安装-CSDN博客
五、测试验证
在安装完torch、torchaudio、torchvision之后
- 输入python
- 输入import torch
- 输入torch.cuda.is_available()
可以参考的验证代码还有:Python – 通过代码验证PyTorch是否安装成功、CUDA是否可用 (hangge.com)
如果以上都OK,但是在jupyternotebook上还用不了,具体可以参考Vscode中使用ipynb文件,解决Python需要安装 ipykernel_需要ipykernel包-CSDN博客
作者:Merci美滋滋