持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第1天，点击查看活动详情

0 写在前面

研一一年断断续续学了些Java，现在研二寻思着还是得学些算法发论文。新手小白入门CV（目标检测），在这里记录一下自己的菜鸡之旅吧！

1 CUDA & pytorch安装

1.1 cuda安装

CUDA 英伟达（NVIDIA）推出的GPU开发环境

根据自己的电脑独显安装对应版本的cuda，如图所示是显卡支持的最高cuda版本。 （小坑：因为开启了电脑管家的桌面管理所以右键不能显示显卡控制面板...）

去官网developer.nvidia.com/cuda-toolki… 下载

第一次安装时出现 NVIDIA安装程序失败 解决：在服务里打开Windows Installer

安装完成后查看cuda版本:cmd中输入 nvcc -V

1.2 pytorch安装

创建一个虚拟环境来安装pytorch 在Anaconda Prompt中输入

conda create -n pytorch python=3.6 创建名为pytorch的虚拟环境
conda activate pytorch 激活虚拟环境

去官网pytorch.org/ 安装对应版本的pytorch，我这里安装的是版本10，可以查看之前版本的安装命令

#CUDA 10.2
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch

2 pycharm配置conda环境

如果这里检测不到可能是系统环境变量没有配置

pycharm终端启用虚拟环境 再切换到当前项目目录即可使用

3 在虚拟环境中安装jupyter

在Anaconda命令行中进入pytorch虚拟环境，执行安装命令

conda install nb_conda
# pip install notebook

再输入jupyter notebook即可在创建的环境中使用jupyter