系统环境:Win10+显卡3090
安装cuda和cudnn:
Windows10系统pytorch、cuda11.0、cuDNN安装
注意事项:
1.查看显卡版本后下载对应版本的cuda和cudnn,链接如下:
cuda下载:https://developer.nvidia.com/cuda-toolkit-archive
cudnn下载:https://developer.nvidia.com/rdp/cudnn-archive
2.安装好cuda后,安装cudnn即将下载的压缩包里的lib,include,bin文件添加到cuda对应路径下。
注意,一台电脑可以安装多个版本的cuda,版本之间的切换只需要将系统环境变量对应版本的路径前置即可。同时,执行Pytorch官方的conda安装命令,也可以在anaconda创建的虚拟环境中搭建pytorch所需对应版本的cuda和cudnn。另外,可以使用conda命令在虚拟环境中单独安装所需cuda和cudnn:例如conda install cudatoolkit=11.2 cudnn=8.1
安装Anaconda:
win10中安装Anaconda
安装Pytorch:
Pytorch官网:https://pytorch.org/
创建虚拟环境:conda create -n env_name python=3.9
注意事项:除非自己有编译需求,否则无需先安装cuda和cudnn,根据官网,只需一条命令即可。因此可以任意选择需要的Pytorch版本,不一定要安装显卡驱动对应的cuda版本,因为Pytorch官网conda命令安装pytorch时会在虚拟环境中重新配置对应的cuda和cudnn。
conda添加清华、阿里源:
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/conda-forge conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --set show_channel_urls yes
conda config --add channels http://mirrors.aliyun.com/anaconda/pkgs/main
conda config --add channels http://mirrors.aliyun.com/anaconda/pkgs/r
conda config --add channels http://mirrors.aliyun.com/anaconda/pkgs/msys2
conda config --set show_channel_urls yes
验证pytorch、cuda、cudnn:
python
import torch
import torchvision
print(torch.__version__)
print(torchvision.__version__)
print(torch.cuda.is_available())
torch.version.cuda
torch.backends.cudnn.is_available()
torch.backends.cudnn.version()
安装Opencv:
conda install -c conda-forge opencv
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-contrib-python pip install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-python
pip添加清华源:
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
安装pycocotools:
pip install cython
(Linux)pip install pycocotools或者(Win)pip install pycocotools-win
Windows和Linux端简单安装pycocotools方法