前言
安装Pytorch,首先需要安装的是torch,做CV的常用的还有TorchVision,还有必不可少的cudatoolkit。
它们之间的版本有着严格的对应关系。
而官网提供的命令没有涵盖我们所需要的一些版本的对应(如下图所示),所以我们只能模仿官网的命令来下载。
![在这里插入图片描述](https://img-blog.csdnimg.cn/533fb48aecae4e9f9497a52383610428.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBA5a6H5a6Z54iG6IKd6ZSm5qCH6LWb5Yag5Yab,size_20,color_FFFFFF,t_70,g_se,x_16)
![在这里插入图片描述](https://img-blog.csdnimg.cn/51733184f76c437fb67d9fc693fcb47e.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBA5a6H5a6Z54iG6IKd6ZSm5qCH6LWb5Yag5Yab,size_20,color_FFFFFF,t_70,g_se,x_16)
我现在使用的是CUDA11.2,那么我要么使用cudatoolkit11.1和cudatoolkit11.2,而不能使用cudatoolkit11.3,因为不能向上兼容,只能向下兼容,升级CUDA很麻烦,我不想这么麻烦。
安装torch和torchvision
torch和torchvision对应关系
pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html
注:conda install 可能找不到源包,那就在虚拟环境中用 pip install.
安装cudatoolkit
配置清华镜像:
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/
anaconda show Zimmf/cudatoolkit
查找cudatoolkit:
anaconda search -t conda cudatoolkit
![在这里插入图片描述](https://img-blog.csdnimg.cn/e248673d03254596871f1ceae30abc35.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBA5a6H5a6Z54iG6IKd6ZSm5qCH6LWb5Yag5Yab,size_20,color_FFFFFF,t_70,g_se,x_16#pic_center) 找到我需要的11.1版本的那一行:
anaconda show Zimmf/cudatoolkit
发现它里面有10.2和11.1的版本,选择我需要的11.1,安装:
conda install --channel https://conda.anaconda.org/Zimmf cudatoolkit=11.1
![在这里插入图片描述](https://img-blog.csdnimg.cn/4173e79249ed4470b4fbd10d5d11026f.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBA5a6H5a6Z54iG6IKd6ZSm5qCH6LWb5Yag5Yab,size_20,color_FFFFFF,t_70,g_se,x_16#pic_center)
|