【教程】虚拟环境与Pytorch安装保姆级教学
- NVIDIA驱动安装
- 虚拟环境
- 创建
- 激活/退出
- 相关库的安装
- Pycharm内设置虚拟环境
- Pytorch安装
- 安装地址
- 可能遇到的问题
- 处理报错
- 安装卡顿
- 测试是否安装完成
- 参考
NVIDIA驱动安装
NVIDIA驱动可在官网进行安装:NVIDIA驱动官网
命令行输入nvidia-smi
可查看cuda版本等信息:
虚拟环境
【声明】 请确保已安装conda与pycharm,可参考博文:【教程】Miniconda+Pycharm安装保姆级教学
创建
进入项目文件位置,命令行输入以下指令,创建虚拟环境
conda create -n 自定义名称 python==3.8.5 ## 创建虚拟环境,定义python版本
补充:进入项目文件主要有两种方式
-
主动打开项目文件,并输入cmd
-
命令行cd到项目文件地址
虚拟环境创建成功后系统返回done
,并出现如下提示:
激活/退出
输入指令conda activate -自定义名称
,激活虚拟环境
首次激活出现如下界面:
PS:可以看到虚拟环境以我们自定义的名称作为开头,并被置于括号内。
若退出则输入指令conda deactivate
即可:
相关库的安装
- 输入
dir
可查看项目文件夹内文件列表:
- 输入指令
pip install -r requirements.txt
进行所需库的安装:
从图中可以看到是通过镜像进行安装,速度还是较快的(因为在安装conda的时候已进行换源)。
Pycharm内设置虚拟环境
- 刚进入pycharm会提示选择系统默认编译环境,若需要使用虚拟环境则跳过就好。跳过后界面右下方会显示
No interpreter
,按下图进行点击添加即可。
- 先前已预设好了conda的虚拟环境,因此在这里选择
conda environment
,并按下图步骤操作即可。
- 选择完
_conda.exe
后点击Load Environments
上传,此后可使用已有环境,亦可创建新环境。
- 此后稍作等待,让系统完成适配即可。
Pytorch安装
安装地址
可在pytorch官网查看相应指令进行安装,官网如下:pytorch官网
针对其余版本(本处以v1.10.0为例):
进入虚拟环境,命令行内输入以下代码即可:
# CUDA 11.3
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3
环境地址为conda安装的地址下的envs
文件夹中。
可能遇到的问题
处理报错
安装过程中可能会遇到如下报错警告:WARNING conda.gateways.disk.delete:unlink_or_rename_to_trash(139)
。经尝试,按照系统相关指令删除指定文件夹内文件即可恢复。具体可参考此篇文章:参考。
安装卡顿
可多次执行上述安装指令
测试是否安装完成
若安装完成命令行返回done
,我们可以通过输入以下代码进行测试:
>>> import torch
>>> print(torch.cuda.is_available())
True
>>> exit()
若返回值为True
,则表示测试成功。
参考
本文参考博文如下:https://blog.csdn.net/ECHOSON/article/details/128130360