
PyTorch 1.11 CUDA 11.3 PyCharm 2022 调试环境配置
5星
- 浏览量: 0
- 大小:None
- 文件类型:DOCX
简介:
本教程详细介绍如何在CUDA 11.3环境下安装和配置PyTorch 1.11及PyCharm 2022,涵盖调试技巧与实战应用。
一个老程序猿要走Pytorch的新路,先搭建一个运行调试环境,踩坑若干若干,那滋味就是一个字=太爽!分享给同路的小伙伴,一些学习成长吧!
涉及的内容包括:
1. 更新显卡驱动至GTX1070 CUDA Version:11.6;
2. 从官网下载对应版本的CUDA Toolkit Archive
3. 安装NVIDIA cuDNN
4. 安装Anaconda3
5. 创建Pytorch_GPU运行的虚拟环境
6. 使用清华镜像快速安装Pytorch GPU版本
7. IDE安装Pycharm,链接Anaconda环境解释器
8. 验证
### Pytorch 1.11_CUDA11.3_Pycharm2022_调试环境搭建
#### 一、更新显卡驱动与确认CUDA版本
为了确保GPU能够正常支持Pytorch,首先需要更新显卡驱动并确认其支持的CUDA版本。
- **步骤1**:通过命令行输入`nvidia-smi`,查看当前显卡驱动和CUDA版本信息。
- **步骤2**:若发现版本不匹配或过旧,则前往NVIDIA中文官网下载最新的显卡驱动程序。安装过程中,请选择“精简”模式以避免错误。
- **步骤3**:重新输入`nvidia-smi`命令,确认更新后的驱动和CUDA版本信息。
#### 二、安装CUDA Toolkit
- **步骤1**:根据所需CUDA版本(此处为11.6),访问NVIDIA开发者官网下载对应版本的CUDA Toolkit Archive。
- **步骤2**:建议使用下载管理器如迅雷进行加速下载。文件名为`cuda_11.6.2_511.65_windows.exe`。
- **步骤3**:按照默认或自定义路径安装CUDA Toolkit,例如 `C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.6`。
- **步骤4**:在系统环境变量Path中添加两个路径:`C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.6bin` 和 `C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.6libnvvp`。
- **步骤5**:重启计算机后,通过命令行工具输入`nvcc --version`,检查CUDA版本是否正确安装。
#### 三、安装NVIDIA cuDNN
- **步骤1**:访问NVIDIA开发者网站下载cuDNN。注册账号以获取下载权限。
- **步骤2**:下载适用于CUDA 11.6的cuDNN版本(例如`cudnn-windows-x86_64-8.4.0.27_cuda11.6-archive.zip`)。
- **步骤3**:解压下载的cuDNN文件,并将其放置于CUDA安装路径下的`cudnn`文件夹中。
- **步骤4**:通过命令行切换至 `C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.6extrasdemo_suite`目录,运行`bandwidthTest.exe`进行测试。如果安装成功,输出结果应为 `Result=PASS`。
#### 四、安装Anaconda3
- **步骤1**:访问清华大学提供的Anaconda镜像站点下载最新版本的Anaconda3。
- **步骤2**:安装完成后,通过Windows开始菜单中的“Anaconda Prompt”打开命令行界面。
- **步骤3**:创建一个新的虚拟环境以安装Pytorch和相关依赖包。例如:`conda create -n Pytorch1.11_GPU python=3.7`。
#### 五、在虚拟环境中安装Pytorch
- **步骤1**:激活创建的虚拟环境:`conda activate Pytorch1.11_GPU`。
- **步骤2**:使用清华镜像快速安装Pytorch GPU版本。命令示例:
```
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch -c conda-forge -c nvidia
```
- **步骤3**:安装完成后,可以通过Python脚本导入Pytorch库进行测试,确认安装无误。
#### 六、安装Pycharm并配置环境
- **步骤1**:下载并安装Pycharm 2022版本。
- **步骤2**:在Pycharm中,配置解释器指向Anaconda3创建的虚拟环境(`Pytorch1.11_GPU`)。
- **步骤3**:创建新的Python项目,并设置工作空间和解释器。
- **步骤4**:验证Pytorch是否能够在Pycharm中正常使用。
通过以上步骤,可以顺利完成基于Pytorch 1.11、CUDA 11.3、Pycharm 2022的GPU调试环境搭建,为后续深度学习项目的开发
全部评论 (0)


