深度学习环境配置Anaconda版本
本次环境配置主要模式是基于 Anaconda+PyTorch(GPU版)+CUDA+cuDNN进行搭建的。CUDA(首先下载两个安装包:CUDA toolkit工具包,cuDNN用于配置深度学习使用,cuDNN其实就是CUDA的一个补丁,专为深度学习运算优化的)
1. 安装Anaconda,最好用镜像源
2.创建虚拟环境
#创建
conda create -n xxx python=3.8
#激活
conda activate xxx
#删除
conda remove -n xxx --all
#查看
conda env list
3.更新NVIDIA显卡驱动
NVIDIA控制面板查看CUDA版本
4.安装CUDA toolkit和cuDNN
toolkit链接版本
CUDNN链接
注:需要登录NVIDIA账号
5.PyTorch安装
镜像源地址
- 清华:https://pypi.tuna.tsinghua.edu.cn/simple
- 中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
- 华中理工大学:http://pypi.hustunique.com/
- 山东理工大学:http://pypi.sdutlinux.org/
- 豆瓣:http://pypi.douban.com/simple/
6.安装pycharm
在官网只需要安装社区版本就可以