新手小白搭建深度学习pytorch环境

qq_27038963

于 2023-06-29 20:26:12 发布

阅读量60

点赞数

文章标签： python 深度学习 pytorch

本文链接：https://blog.csdn.net/qq_27038963/article/details/131463285

版权

新手小白学习安装pytorch框架，参照网上各类帖子，血泪流干也没安装成功，最后发现很多帖子思路都是错的，在此总结我的思路：

1. 安装Anaconda：去官网https://www.anaconda.com/下载安装即可。

2. 配置pytorch：

打开电脑的英伟达控制面板，记下CUDA的版本号，我的理解是这个版本号表示目前的显卡驱动所能支持的最高版本CUDA。

打开pytorch官网：https://pytorch.org/，根据选项卡配置自己的pytorch环境，package一般选pip，language一般选python，最最最最最重要的是CUDA一定要选不超过上图提到的版本（没有的话可以在箭头处找老版本），复制圆框内的命令在cmd中执行。

若遇到“pip 不是内部或外部命令”的问题，参考https://blog.csdn.net/weixin_43252521/article/details/120212381

若遇到“ERROR: Exception: Traceback (most recent call last)”，参考https://blog.csdn.net/inside802/article/details/102646240

3. cmd执行结束，深度学习环境也就安装完成了。此时pytorch、CUDA及相匹配的cudnn都有了，并不需要各个帖子说的那样各个包分别下载安装。打开anaconda中的Spyder，输入以下语句执行可以查到pytorch、CUDA及cudnn版本号：

import torch

print(torch.__version__)

print(torch.version.cuda)

print(torch.backends.cudnn.version())

运行结果：

2.0.1+cu117
11.7
8500

关注