windows + anaconda + pycharm + nvidia + cuda + cudnn + pytorch 安装教程
要学pytorch了耶,这不得安装一把子环境?
查看显卡信息
操作:开始 —— 右键 —— 设备管理器 —— 显示适配器。
查看cuda版本
操作:右下角NVIDIA —— 右键 —— NVIDIA控制面板 —— 系统信息 —— 组件。
安装cuda
去CUDA官网下载cuda10.2版本。
下载后双击exe文件,更改目录,解压。
解压完成后会自动进入安装流程。
按照安装提示默认安装即可。
安装中。
检查是否安装成功。
nvcc -V
成功嘞!
安装cudnn
去CUDNN官网下载cuda10.2版本对应的cudnn。
下载之后,解压缩,将CUDNN压缩包里面的bin、include、lib文件直接复制到CUDA的安装目录下(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2)。
这里会遇到管理员权限问题。
在”为所有当前项目执行此操作“前点击对号再点击继续。
配置cuda环境变量
已经有两个了。 还要添加三个。
一开始找不到CUDA Samples,但是开始菜单可以看到。
打开即可。 后来发现,系统变量的PATH里有bin和libnvvp就可以了。
配置cudnn环境变量
安装pytorch
在conda里面查看python版本。 在conda里面创建虚拟环境。
wxm_pytorch是虚拟环境名字。
conda create -n wxm_pytorch python=3.7
点击y。
激活虚拟环境。
conda activate wxm_pytorch
进入pytorch官网选择配置项。
conda安装
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
输入y,等待安装。
下好啦。
下错了。
先删除这个失败的虚拟环境。
重来。
pip安装
pip3 install torch==1.10.2+cu102 torchvision==0.11.3+cu102 torchaudio===0.10.2+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html
下好了。
测试一下:
开心心。
看到torch了。
第一个DEMO
终于!!!
|