> 文章列表 > 如何在Anaconda下安装pytorch(conda安装和pip安装)

如何在Anaconda下安装pytorch(conda安装和pip安装)

如何在Anaconda下安装pytorch(conda安装和pip安装)

前言

文字说明

本文中标红的,代表的是我认为比较重要的。        

版本说明

  • python环境配置:jupyter的base环境下的python是3.10版本
  • CUDA配置是:CUDA11.6。
  • 目前pytorch官网提示支持的版本是3.7-3.9

 本文主要用来记录自己在安装pytorch中出现的问题,同时希望我也能够帮助大家解决一些共同遇到的问题。

如何查看自己的Anaconda环境下的pytorch

  • 第一步,找到自己的Anaconda Prompt

  •  第二步,打卡它

  •  第三步,构建新环境(为了让python版本降低下来,不然python3.10下的torch安装会出现torchvision的安装不成功的情况)

        python程序的运行很依赖环境,因为在base环境下的python已经是python3.10了,所以尝试创建一个新的环境,用来存放pytorch。【突然发现自己没有尝试去,把base环境下的python版本给降低,待会去试一下】

        如何构建虚拟环境:

        1.如何查看自己的Anaconda的环境。输入

conda env list

如何不在Anaconda环境下安装pytorch

第一步 win + r 

第二步 cmd

第三步 调用如下命令

pip3 install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple

 第四步 pip list检查安装

 第五步 IDE环境下运行

 经过上面几步,我们的pytorch库下的torch,torchvision,torchaudio就安装成功。