一、背景
前提是已经安装好python环境。
二、配置过程
1.cuda安装
官网下载合适版本的Installer,双击打开,跟着指引一步步操作即可。
注意,如果出现NVIDIA程序安装失败,可以在"自定义安装选项"步骤处取消勾选Visual Studio Integration和Nsight VSE选项,方可成功安装。
2.cudnn安装
官网下载版本合适的并解压,将文件下的bin,include,lib三个文件夹都copy到cuda的安装目录下。
图一是cudnn解压后的文件情况,图二是把三个文件夹拷贝到cuda的安装目录下。
3.环境变量的配置
(1)检查系统变量里是否有cuda的安装根目录,如图,一般安装完都会有,没有的话自己加上。
(2)在系统变量-PATH中添加以下几项,注意该改成自己安装路径的都改一下。
4.pytorch的安装
在官网上找到对应版本,复制命令下载即可。尤其注意CUDA版本是否匹配!
5.jupyter notebook的安装
我用的是anaconda-虚拟环境,直接打开anaconda prompt切入虚拟环境,执行命令conda install jupyter notebook即可。记得在安装之前配置国内镜像源,否则速度会很慢!
6.jupyter notebook的启动
在Anaconda prompt的虚拟环境中直接输入命令jupyter notebook
打开后应该是如此界面:
7.jupyter notebook对应虚拟环境
右上角New,新建一个Notebook程序,选择你想要的虚拟环境。
Kernel处也做相应的选择。
9.jupyter notebook跑Hello World程序
先import torch试试,看是否报错?如图,报错no module named torch的话,就意味着需要安装torch。
介绍一种最为便捷的方法:在Anaconda Prompt对应的虚拟环境中执行命令pip install torch
后面如果遇到类似no module named xxx的报错,应该也是同样的解决办法。
conda list可以查看是否安装成功,即查看该虚拟环境里所有的包。
再返回jupyter notebook,执行import torch不出现报错,hello world程序跑成功了!
三、心得体会
配环境就是熟能生巧的过程,之前装过一次,第二次装便轻松许多。记录下来做备忘录。