主要可以解决用conda下载pytorch慢以及cuda版本不匹配的问题
研究生新生一名,以前从来没接触过深度学习、计算机视觉,但是最近导师想换到这个方向,作为新生也只能硬着头皮上,下面记录一下自己今天装pytorch过程,有不妥之处请不要喷,指出来我一定虚心接受。
1. 首先确定自己电脑上的GPU以及CUDA版本
这个没什么好说的,找到设备管理器–显示适配器–帮助–组件,就能看到自己电脑对应的版本。
2. 到pytorch官网找到对应的版本
比如我是cuda11,找到对应的。关键是不要选conda下的安装命令,选择pip命令安装。亲试用conda命令安装特别特别慢,而且装了几次都没装成功,pip一装就成功了。
如果是在cmd里面,需要先进入conda虚拟环境
conda info --envs # 查找conda环境
conda activate pytorch_env # 激活自己的环境
pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
静静等到即可,正常情况下不会出任何问题,下载速度相对用conda命令快非常多
如果要装opencv直接用以下命令即可
conda install opencv
有可能会报错
[Errno] Permission denied:.........
看看自己是不是文件夹未关闭(一般不是这个问题),如果不是,将cmd窗口关闭之后重试命令即可安装成功。