pytorch和pycocotools的安装

最新推荐文章于 2025-03-26 15:23:11 发布

敏儿要努力

最新推荐文章于 2025-03-26 15:23:11 发布

阅读量3.8k

点赞数 17

文章标签： pytorch 深度学习 python 机器学习

本文链接：https://blog.youkuaiyun.com/nshk1031/article/details/122132609

版权

记录一下自己安装pytorch 和 pycocotools

一，Anaconda 安装
二，pytorch 安装
三，安装 pycocotools

一，Anaconda 安装

参考博客
首先保证你有python3.6/3.7的环境,可以在官网或者清华镜像网站上下载Anaconda3.
打开Anaconda Prompt,出现以下界面表示安装成功:
在这里插入图片描述
查看python版本号

二，pytorch 安装

1，先是创建pytorch环境
其中pytorch是环境名称；python=3.7是python版本
在这里插入图片描述
出现 Proceed ([y]/n)? 时输入y即可

2，为了加快后续安装的速度，防止软件太大下载中断。我们可以加上清华源：
在这个base命令窗口添加

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
# reference
# https://mirror.tuna.tsinghua.edu.cn/help/anaconda/

或者pytorch的源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

3，输入activate pytorch，左边的环境就从base（基本环境）变为pytorch环境。
在这里插入图片描述
4，win+r ，输入cmd , 然后输入 nvidia-smi 查看自己英伟达的版本号。具体参考

5，打开pytorch官网
若安装最新版本，则在下面界面，根据自己电脑配置选择合适的选项，要保证你的电脑上的CUDA版本不小于你选择的版本。使用清华源或pytorch源的话则把后面的 -c pytorch 删掉。
在这里插入图片描述
下图表示的是pytorch1.10版本（注意是在pytorch环境下）

若安装指定版本的pytorch，则在这个网址
比如我要在windows上安装pytorch1.6 ，则
在这里插入图片描述
同样把 -c pytorch 删掉

6，验证安装
python
import torch
torch.cuda.is_available()
如下图所示则表示安装成功：

三，安装 pycocotools

Windows下 pip install pycocotools-windows(不需要额外安装vs))
因为我已经安装好了，所以显示already satisfied
在这里插入图片描述
四，总结
自己在安装的时候差不多花了一个星期的晚上，总是出错。比如之前没有加上清华源，总是安装中断，有时候软件太大、安装时间过长，电脑息屏后也会安装中断，后来就在电脑旁守着，最后终于安装好了。