PyTorch环境的配置及安装

ゞ浴缸里的玫瑰 2023-09-28 10:22 128阅读 0赞

1、安装Anaconda  
[https://repo.anaconda.com/][https_repo.anaconda.com]  
安装相应的版本，我这里安装的是Anaconda3-5.2.0-Windows-x86\_64.exe

![在这里插入图片描述][fd1e9db6d8ae40acb0db7090a925a829.png]  
![在这里插入图片描述][adb4cc36e7e947ecb515846d5f919e79.png]  
![在这里插入图片描述][0a88515af3af403fb022365b92465e99.png]  
![在这里插入图片描述][ae2cb5a89bc9472a81c5a7b750ffbe2f.png]

2、显卡准备  
深度学习离不开显卡（PyTorch支持英伟达显卡）  
检查显卡驱动是否正常安装：正常显示表示，驱动正常安装。  
![在这里插入图片描述][9a3e1343ee344ff2b5a6a0cdc768edef.png]

3、管理环境

3.1、  
![在这里插入图片描述][cc3a6e96aa6c4074b0e259c354bfdd43.png]  
![在这里插入图片描述][97668c8fe83446bfa56a290de9417463.png]

3.2、初始环境（base） (初始的屋子)  
下面我们创建另一个环境（pytorch）:

conda create -n pytorch python=3.6

![在这里插入图片描述][176a1e724c804f2fb878db2148ee4e86.png]

3.3、激活环境(pytorch):

conda activate pytorch

![在这里插入图片描述][9b4c163c5518447a83758eddf04041db.png]

3.4、查看这个环境（屋子）中有哪些包

pip list

![在这里插入图片描述][bd7c0d1d41e84d5181eb31df66770c5a.png]  
很明显，上面的包中没有pytorch包，所以我们要安装pytorch

3.5.安装pytorch  
3.5.1、进入pytorch官网：[https://pytorch.org/][https_pytorch.org]  
英伟达显卡，如下操作：  
![在这里插入图片描述][590871a010ff409e94923d80f766825d.png]

选好后，复制运行指令：

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

3.5.2、查看驱动版本是否符合conda要求  
命令窗口 输入`nvidia-smi`，查看驱动版本。  
conda仅支持驱动版本号大于396.26  
![在这里插入图片描述][eb1533607a7f491b89b2635c13640c52.png]

3.5.3,、  
在环境(pytorch)中运行3.5.1复制的指令

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

![在这里插入图片描述][e07dfb46142d4cd89391e22072e40143.png]  
再查看一下安装的包有哪些

pip list

![在这里插入图片描述][31083d816fae44138df8776d1178d638.png]

很明显，pytorch包成功安装

3.6、检查pytorch是否安装成功  
在pytorch环境中 输入 `python`  
输入`import torch`若没有报错，正常显示，则pytorch成功安装。  
![在这里插入图片描述][5da5c003af02442385d0f20eab34a1fd.png]

3.7、检查pytorch是否可用使用你电脑的GPU  
输入`torch.cuda.is_available()`  
如果显示True，则你的这台计算机的GPU可以给pytorch使用  
![在这里插入图片描述][0773e15071e64390b8e3c046cfcdc827.png]

至此，大功告成。

4.总结

*  创建conda环境，指定python版本
 *  激活环境使用
 *  安装pytorch  
    ![在这里插入图片描述][677ae36de0e144c289931d2b940270d2.png]

[https_repo.anaconda.com]: https://repo.anaconda.com/
[fd1e9db6d8ae40acb0db7090a925a829.png]: https://img-blog.csdnimg.cn/fd1e9db6d8ae40acb0db7090a925a829.png
[adb4cc36e7e947ecb515846d5f919e79.png]: https://img-blog.csdnimg.cn/adb4cc36e7e947ecb515846d5f919e79.png
[0a88515af3af403fb022365b92465e99.png]: https://img-blog.csdnimg.cn/0a88515af3af403fb022365b92465e99.png
[ae2cb5a89bc9472a81c5a7b750ffbe2f.png]: https://img-blog.csdnimg.cn/ae2cb5a89bc9472a81c5a7b750ffbe2f.png
[9a3e1343ee344ff2b5a6a0cdc768edef.png]: https://img-blog.csdnimg.cn/9a3e1343ee344ff2b5a6a0cdc768edef.png
[cc3a6e96aa6c4074b0e259c354bfdd43.png]: https://img-blog.csdnimg.cn/cc3a6e96aa6c4074b0e259c354bfdd43.png
[97668c8fe83446bfa56a290de9417463.png]: https://img-blog.csdnimg.cn/97668c8fe83446bfa56a290de9417463.png
[176a1e724c804f2fb878db2148ee4e86.png]: https://img-blog.csdnimg.cn/176a1e724c804f2fb878db2148ee4e86.png
[9b4c163c5518447a83758eddf04041db.png]: https://img-blog.csdnimg.cn/9b4c163c5518447a83758eddf04041db.png
[bd7c0d1d41e84d5181eb31df66770c5a.png]: https://img-blog.csdnimg.cn/bd7c0d1d41e84d5181eb31df66770c5a.png
[https_pytorch.org]: https://pytorch.org/
[590871a010ff409e94923d80f766825d.png]: https://img-blog.csdnimg.cn/590871a010ff409e94923d80f766825d.png
[eb1533607a7f491b89b2635c13640c52.png]: https://img-blog.csdnimg.cn/eb1533607a7f491b89b2635c13640c52.png
[e07dfb46142d4cd89391e22072e40143.png]: https://img-blog.csdnimg.cn/e07dfb46142d4cd89391e22072e40143.png
[31083d816fae44138df8776d1178d638.png]: https://img-blog.csdnimg.cn/31083d816fae44138df8776d1178d638.png
[5da5c003af02442385d0f20eab34a1fd.png]: https://img-blog.csdnimg.cn/5da5c003af02442385d0f20eab34a1fd.png
[0773e15071e64390b8e3c046cfcdc827.png]: https://img-blog.csdnimg.cn/0773e15071e64390b8e3c046cfcdc827.png
[677ae36de0e144c289931d2b940270d2.png]: https://img-blog.csdnimg.cn/677ae36de0e144c289931d2b940270d2.png