Window10YOLO:环境搭建3

En_嗯

已于 2023-03-24 10:34:20 修改

阅读量366

点赞数 1

分类专栏： # YOLO 文章标签： python pytorch 深度学习

于 2023-03-16 11:08:02 首次发布

本文链接：https://blog.csdn.net/weixin_46355605/article/details/129571513

版权

YOLO 专栏收录该内容

4 篇文章 1 订阅

订阅专栏

本文详细介绍了如何下载和安装CUDA11.3以及对应版本的PyTorch，包括通过Anaconda创建环境、从特定网址下载whl文件进行离线安装，并提供了清华源的配置以加速下载。关键步骤包括选择正确的CUDA和Python版本、创建Anaconda环境、放置whl文件及激活环境后安装PyTorch。

摘要由CSDN通过智能技术生成

预备工作

查找下载的CUDA与pytorch的对应版本。

1、官网下载numpy（torchvision）：

https://pytorch.org/get-started/previous-versions/（新版）
如果想要寻找以前的版本就进入下面的网址
https://pytorch.org/get-started/previous-versions/#linux-and-windows-4（旧版）
在这里插入图片描述

准备Anaconda命令

conda install pytorch torchvision torchaudio cudatoolkit=11.3

因为我的CUDA是11.3，所以选择这个下载命令

2、下载pytorch

PyTorch 版本	CUDA 环境
0.4.1、1.2.0、1.4.0、1.5.0(1)、1.6.0、1.7.0(1)	9.2
1.2.0、1.1.0、1.0.0(1)	10.0
1.4.0、1.5.0(1)、1.6.0、1.7.0(1)	10.1
1.5.0(1)、1.6.0、1.7.0(1)、1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0、1.12.0(1)	10.2
1.7.0(1)	11.0
1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0	11.1
1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0、1.12.0(1)	11.3
1.11.0	11.5
1.12.0(1)	11.6

下载PyTorch

本篇文章只推荐离线安装：https://download.pytorch.org/whl/torch_stable.html
因为新手在官网安装容易出错，加上可能找不到对应的版本，这里的cu：cuda版本；cp：Python版本；torch：torch版本，选择适合自己的系统是Window，点击链接下载即可
我下载的如图所示：
在这里插入图片描述
下载位置随便，放在一个好找的地方，后面会用
下载好后，进行离线安装：打开Anaconda Prompt

安装PyTorch

1、首先输入下面代码进行创建环境：

代码中python310是你创建的环境文件夹名字，数字3.10是python环境的版本，按照你的配置更改

conda create -n python310 python=3.10

创建完的效果图：
在我的这个地址下：D:\11Anaconda\envs
在这里插入图片描述

2、把刚才下载好的pytorch文件，放到创建的python310文件夹里面，如下图：

在这里插入图片描述

3、切换一下清华源操作如下：重新打开Anaconda Prompt输入如下（防止错误）

conda activate python310
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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

4、输入下面代码激活环境,使用一开始准备的Anaconda命令，安装pytorch，下载成功会显示successful installed

conda activate python310
conda install pytorch torchvision torchaudio cudatoolkit=11.3
cd D:\11Anaconda\envs\python310
D:
pip install torch-1.11.0+cu113-cp310-cp310-win_amd64.whl

效果如下图所示：
在这里插入图片描述

以上pytorch就安装完毕了，需要注意就是创建环境时也就是conda create -n python310 python=3.10 这个语句最后python的版本要与下载的pytorch的版本对应，之后将这个对的版本，放进设定这个python版本的环境文件夹中，也就是如下图：
在这里插入图片描述
如果上面操作不顺利，就再切换完清华源之后再操作一下conda create -n python310 python=3.10,之后的按照步骤来