可以先参考这个链接写得十分详细。
Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch详细教程_windows cudacudnn配置
1.anaconda安装
浏览器搜索anaconda下载。Free Download | Anaconda
自己先在D盘(哪个盘都可以不要放C盘)新建一个文件夹anaconda,安装路径选择这个。
直接点击Finish,点击后跳出的网页不用管,可以关掉。
后面主要是使用 Anaconda Prompt
在电脑左下角的开始,所有应用anaconda文件下可以找到
使用 Anaconda Prompt创建conda环境
conda create -n pytorch python=3.11 (pytorch是创建的环境名称,可以自行修改;3.11是python的版本)
2.pytorch安装
需要先下载CUDA和cuDNN(自己电脑上的CUDA版本查看)。
注意:在安装CUDA是要先新建两个文件夹,第一次extraction path选择时的文件夹会在提取完成后自动删除,第二的文件夹是安装路径。
下载CUDA
cuDNN下载安装 需要注册账号
浏览器搜索cudnn,找到对应CUDA版本的cuDNN进行下载
浏览器搜索pytorch
选择自己的版本(需要先查看cuda和cudnn的版本,两者版本要相互对应)
进入在前面用Anaconda新建的环境(我这里新建的环境名称是pytorch)里复制粘贴红框内的指令
3.Pycharm安装
浏览器搜索下载社区版Download PyCharm: Python IDE for Professional Developers by JetBrains
下滑找到PyCharm Community Edition
到这里需要的软件就安装好啦,接下来环境配置可以参考以下视频(6:14开始)