前言
文字说明
本文中标红的,代表的是我认为比较重要的。
版本说明
- 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就安装成功。