驱动更新
首先检查机器当前驱动版本以及所支持的cuda版本。
桌面右键点击NVIDIA控制面板
可以看到显卡型号为1070,驱动版本为456.71。
点击帮助->系统信息->组件
可以看到当前驱动支持cuda版本为11.1,即11.1以下均可,如果版本不够,需要到官网下载最新的驱动:
anaconda安装
下载链接(各个版本):repo.continuum.io/archive/
下载后直接默认安装。
安装成功后打开cmd输入conda list
检查是否安装成功。
vscode2017安装
自行找安装程序安装,只需要注意选择安装配件的时,选择科学计算和python即可。同时建议默认安装在c盘即可。(删除)
cuda安装
安装cuda前注意版本关系:
cuda下载地址:developer.nvidia.com/cuda-toolki…
下载后点击安装,建议别改安装地址
然后下载cudnn:developer.nvidia.com/cudnn
下载cudnn后解压可以看到几个文件夹和一个txt说明
接下来要做的就是进入到cuda的安装目录(默认安装):
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
可以看到其中也有和cudnn中相同的几个文件夹,接下来只需要将cudnn中几个文件夹的内容拷贝到cuda中对应的文件夹即可(注意lib文件夹中还有一个x64文件夹,需要将x64文件夹中内容拷贝到对应文件夹)
安装完成后配置一下环境变量:
计算机上右键,打开属性->高级系统设置->环境变量,可以看到系统中多了CUDA_PATH和CUDA_PATH_V10_1两个环境变量,接下来,还要在系统中添加以下几个环境变量:
CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1(这是默认安装位置的路径)CUDA_LIB_PATH = %CUDA_PATH%\lib\x64CUDA_BIN_PATH = %CUDA_PATH%\binCUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64
然后在系统变量 PATH 的末尾添加:%CUDA_LIB_PATH%;%CUDA_BIN_PATH%;%CUDA_SDK_LIB_PATH%;%CUDA_SDK_BIN_PATH%
再添加如下4条(默认安装路径):
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1\common\lib\x64;C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1\bin\win64;
配置完成后,我们可以验证是否配置成功,主要使用CUDA内置的deviceQuery.exe 和 bandwithTest.exe:
打开cmd,进入到cuda安装目录下的 ...\extras\demo_suite,然后分别执行。.\bandwidthTest.exe和.\deviceQuery.exe
得到 result = pass即可。
pytorch安装
网址:pytorch.org/
根据提示选择conda安装需要版本的pytorch,最后会得到一条安装。如果你有科学上网工具,可以直接运行该命令安装,否则先提前设置一下清华源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
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
然后安装pytorch时还需要注意:删除最后的 -c pytorch,否则还是默认的官网源。
注:
这里安装pytorch出现了一些曲折,按照上述conda方法安装的pytorch用不了gpu,通过打印torch.cuda.is_available为false,一直以为是版本问题,装了很久也不行,最后尝试了使用pip方法安装:
pip install --default-timeout=1000 -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.5.1+cu101 torchvision==0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html
打印torch.__version__输出:1.5.1+cu101
再打印torch.cuda.is_available()终于为true!
IDE安装
ide个人喜欢用vscode2019,安装后建议修改一下setting.json文件:
"terminal.integrated.shell.windows": "C:\\WINDOWS\\System32\\cmd.exe"
加上这句后,默认会使用cmd而不是poweershell,这样的话就不会出现切换不了anaconda环境的问题了。