深度学习环境pytorch搭建,保姆级别的极简安装
1 Anaconda软件安装
链接: link.
安装完成后选择Anaconda Prompt命令窗口打开,进行相关库的安装
2 pytorch环境搭建
2.1 nvidia控制面板查看系统组件信息
- 注意:上图中的cuda11.7代表你后续安装cudatoolkit的版本不能超过11.7
2.2 打开Anaconda Prompt命令窗口
- 注意: 默认是base环境下,可以选择在该环境下直接安装相关库,也可以重新激活新的虚拟环境,为保证各个版本库之间不相互干扰,建议激活一个新的环境。
2.3 框架搭建
- 配置镜像源
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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/ 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/
- 激活虚拟环境,torch1.12为虚拟环境名称(随意命名),python版本看自己需求
conda create -n torch1.12 python=3.8
- pytorch官网查找对应版本
链接: link.
注意:30系以上的卡建议选择cuda11.0以上的版本,这里以pytorch1.12为例,可以选择cuda 11.3版本进行安装,不用选择-c pytorch(代表默认官方镜像源安装,会很慢)conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3
- 验证pytorch是否可用(tru)
python import torch torch.cuda.is_available()
2.4 pycharm安装
- 链接: link.(可以自行去官网下载)
- 环境加载
File->settings->Project Interpreter(点击左上角的设置)
Add选择存在的环境,添加环境路径
不知道环境路径的话可以打开Anaconda Prompt命令窗口执行如下指令conda env list
找打pt1.12环境下的python.exe加载的pycharm中,确认后即可加载所有环境。
2.5 pip源的更换(看自己需求可弄可不弄,可解决pip安装过慢的问题)
- C:\Users\26274\目录下新建文件夹pip(C:\Users\26274\ 这个目录也是.condarc文件所在的目录)
- pip文件夹新建pip.txt,复制以下内容配置清华源
[global] index-url = https://pypi.tuna.tsinghua.edu.cn/simple
- 将pip.txt文件改为pip.ini
- 后续在Anaconda Prompt命令窗口通过pip install 安装相关库时,会默认使用清华源加速。