Geforce RTX 3080 windows10 anaconda环境下的pytorch torchvision torchaudio安装

主要可以解决用conda下载pytorch慢以及cuda版本不匹配的问题

研究生新生一名,以前从来没接触过深度学习、计算机视觉,但是最近导师想换到这个方向,作为新生也只能硬着头皮上,下面记录一下自己今天装pytorch过程,有不妥之处请不要喷,指出来我一定虚心接受。

1. 首先确定自己电脑上的GPU以及CUDA版本

这个没什么好说的,找到设备管理器–显示适配器–帮助–组件,就能看到自己电脑对应的版本。
Geforce RTX 3080 windows10 anaconda环境下的pytorch torchvision torchaudio安装_第1张图片

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命令快非常多
Geforce RTX 3080 windows10 anaconda环境下的pytorch torchvision torchaudio安装_第2张图片
如果要装opencv直接用以下命令即可

conda install opencv

有可能会报错

[Errno] Permission denied:.........

看看自己是不是文件夹未关闭(一般不是这个问题),如果不是,将cmd窗口关闭之后重试命令即可安装成功。

你可能感兴趣的:(anaconda,深度学习,python)