Windows 下 PyTorch 入门深度学习环境安装与配置 GPU 版

1.确定自己的硬件信息，确定电脑有英伟达 (NVIDIA)显卡

????????在任务栏上右键打开任务管理器

2.下载安装 Anaconda

????????（建议安装迅雷下载，同时浏览器添加扩展 “迅雷Chrome支持”）

????????https://www.anaconda.com/

????????https://repo.anaconda.com/archive（历史版本）

????????双击安装包

????????换一下路径尽量不要安装在C盘

????????安装完了，双击这个运行

????????刚开始可能速度有点慢多等等，有更新提示就更新

????????有如图所示界面就成功了

????????去寻找安装目录，把anaconda3中envs文件夹，安全权限修改全部允许

3.利用 conda 或者 pip 安装 PyTorch?

????????确定自己安装的CUDA版本与GPU关系，安装GPU版本 PyTorch

????????打开软件的黑窗口

Anaconda命令行，查看虚拟环境
conda env list
创建虚拟环境
conda create -n test python=3.9
开启
conda activate test
关闭
conda deactivate
移除虚拟环境
conda remove -n test --all

指定路径方式
D:
conda create --prefix=D:/Ruanjian/anaconda3/envs/test python=3.9
conda remove --prefix=D:/Ruanjian/anaconda3/envs/test --all
conda activate D:\Ruanjian\anaconda3\envs\test
conda deactivate

????????确定自己显卡的算力
????????确定自己的可选择CUDA Runtime Version
????????确保自己的CUDA Driver 版本>= CUDA Runtime 版本

nvidia-smi

????????无法确保版本对应关系，就无脑升级到最新

????????安装显卡驱动最新版本
????????https://www.nvidia.cn/Download/index.aspx?lang=cn

? ? ? ? 按自己的电脑配置来

????????打开软件的黑窗口

conda info --env
activate test

????????添加清华镜像源：

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

????????输入命令行：

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c nvidia ?

????????应该会下载好久一直等就行

????????再等一会......

????????下载完成

????????验证是否安装成功

import torch
torch.cuda.is_available()

3.配置 PyCharm

????????本文章仅供学习交流使用，如有侵权，联系删除