文章目录
- 一.pytorch 1.10版本安装教程
一.pytorch 1.10版本安装教程
PyTorch官网
-
pytorch与cuda版本关系
官网 -
从官网选择自己对应的conda,python,cuda版本,复制conda命令
-
在windows搜索框中搜索CMD,选择以管理员身份运行
-
使用conda新建虚拟环境pytorch
conda create -n pytorch python=3.8
-
激活新建的虚拟环境
conda activate pytorch
-
更换镜像下载源
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 --set show_channel_urls yes conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ 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/
-
运行pytorch官网上的代码,删除后面的-c pytorch
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3
conda install pytorch torchvision torchaudio cudatoolkit=11.3
-
等待安装完成
-
其他深度学习包安装
pip install scikit-image -i https://pypi.mirrors.ustc.edu.cn/simple/ pip install pillow -i https://pypi.mirrors.ustc.edu.cn/simple/ pip install opencv-python -i https://pypi.mirrors.ustc.edu.cn/simple/ pip install tqdm -i https://pypi.mirrors.ustc.edu.cn/simple/ pip install matplotlib -i https://pypi.mirrors.ustc.edu.cn/simple/ pip install wandb -i https://pypi.mirrors.ustc.edu.cn/simple/ pip install tensorboard -i https://pypi.mirrors.ustc.edu.cn/simple/ pip install pycuda -i https://pypi.mirrors.ustc.edu.cn/simple/