Pytorch最新安装教程(anaconda\python\CUDA\pytorch)

 目录

一、写在前面

二、Anaconda/Python安装配置

三、CUDA配置准备

四、Pytorch安装

五、测试验证


一、写在前面

如果你遇到下面几个问题,那么这篇教程就非常适合你:

  1. 搜到的已有教程太多,难以分辨
  2. cuda/cudatoolkit/cudnn和pytorch/torchaudio/torchvision对应版本不会选
  3. pytorch最新更新的版本里没有conda install链接,搜到的教程都没有提到这一点
  4. pytorch安装太慢,镜像源也不行

以上教程是基于已有教程和自己经验的总结,分为四个部分,如果帮助请多多点赞和转发。

二、Anaconda/Python安装配置

这一步主要安装VScode/Anaconda/Python三个,链接如下:

  • VScode链接:Download Visual Studio Code – Mac, Linux, Windows
  • Anaconda链接:Download Now | Anaconda
  • Python链接:Download Python | Python.org (尽量选最新且security的)
  • 下载完成之后,我参考的是如下的安装教程,十分详细,按步骤照做就行。

    全网最简约的Anaconda+Python3.7安装教程Win10(百分百成功)_anaconda python3.7-CSDN博客

    三、CUDA配置准备

    这一步是CUDA的相关配置,主要是cudatoolkit和cudnn的安装,这一步关键是CUDA的版本对应(千万不能下载高于你电脑适配版本的CUDA),详细教程参考如下:

    https://www.python.org/downloads/

  • cudatoolkit链接:CUDA Toolkit 12.8 Downloads | NVIDIA Developer
  • cudnn链接:cuDNN 9.7.1 Downloads | NVIDIA Developer
  • 四、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之后

    1. 输入python
    2. 输入import torch
    3. 输入torch.cuda.is_available()

    可以参考的验证代码还有:Python – 通过代码验证PyTorch是否安装成功、CUDA是否可用 (hangge.com)

    如果以上都OK,但是在jupyternotebook上还用不了,具体可以参考Vscode中使用ipynb文件,解决Python需要安装 ipykernel_需要ipykernel包-CSDN博客

    作者:Merci美滋滋

    物联沃分享整理
    物联沃-IOTWORD物联网 » Pytorch最新安装教程(anaconda\python\CUDA\pytorch)

    发表回复