windows系统电脑安装深度学习教程

  • windows系统电脑安装深度学习教程
  • 1、Anaconda集成环境
  • 2、下载Cudnn和CUDA
  • 3、配置torch环境


windows系统电脑安装深度学习教程

对于深度学习来说,安装pytorch和tensorflow是基础中的基础,毕竟无法搭建环境就无法进行训练程序,我也是首先对window安装好pytorch和tensorflow,使用一段时间后采继续转向Ubuntu系统的学习使用。现在市场上用的比较多的深度学习也是pytorch和tensorflow,我这次安装的教程是pytorch==1.2.0。右键我的电脑,管理,设备管理器,显示适配器。

Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_windows系统

1、Anaconda集成环境

Anaconda集成环境的安装路径可以不放在C盘默认位置,可以更改到其他盘符,而且大家安装环境之前先把360,电脑管家等杀毒软件卸载了啊。

Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_深度学习_02


Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_人工智能_03


Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_python_04


点击Install就可以对Anaconda3进行安装。

2、下载Cudnn和CUDA

我们这里使用的是torch==1.2.0,官方推荐是Cuda版本10.0,对应的cudnn是7.4.1.5,大家都不用下载,都已经为大家下载好了啊。Cuda和cudnn的安装就在系统的安装默认路径就好了,不必修改。

Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_人工智能_05


下一步

Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_CUDA_06


Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_深度学习_07

同意并继续,后面继续

Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_CUDA_08


Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_深度学习_09


继续下一步

Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_CUDA_10


点击NEXT就可以了啊。

安装完成后可以在自己电脑的这个位置找到安装目录。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

然后大家把Cudnn的内容解压

Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_深度学习_11


将解压好的这几个文件,全部复制粘贴到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0,用于替换。

Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_CUDA_12


一直点击继续就可以了啊。

3、配置torch环境

电脑Win+R启动cmd,在命令提示符中输入以下命令。

Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_python_13


1、创建环境

Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_python_14


出现提示按y即可。

2、激活环境

Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_windows系统_15


Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_windows系统_16


说明环境搭建成功

3、安装

Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_python_17


在文件夹的这个位置输入cmd

Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_windows系统_18


Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_深度学习_19


Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_深度学习_20


在命令行输入

Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_python_21


pip install torch-1.2.0-cp36-cp36m-win_amd64.whl -i https://mirrors.aliyun.com/pypi/simple/

然后输入

Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_windows系统_22


pip install torchvision-0.4.0-cp36-cp36m-win_amd64.whl -i https://mirrors.aliyun.com/pypi/simple/

安装完成后,我们的环境就搭建好了

Windows如何设置pytorch的PYTORCH_CUDA_ALLOC_CONF pytorch在windows_人工智能_23


此后大家需要什么库就可以安装什么库

pip install 库名 -i https://mirrors.aliyun.com/pypi/simple/

比如:pip install numpy -i https://mirrors.aliyun.com/pypi/simple/

这样我们的深度环境pytorch==1.2.0就安装好了,无论大家是pycharm还是VS code都可以直接导入我们这个环境,对于后面的Ubuntu系统深度学习的安装也是这样,有什么问题大家也都可以来跟我交流。

这些是百度网盘链接:

链接:https://pan.baidu.com/s/1S-7YZ-7jESigY-1sW8KYzA

提取码:cgej

最后推荐b站视频给大家:BV1Ui4y147un