Windows上Anaconda配置cuda、pytorch、opencv

一、安装Anaconda

        前往Anaconda官网下载安装包并安装。

默认情况下，Anaconda会将Python解释器和常用的Python库安装在C盘的Anaconda3目录下。安装完成后，打开Anaconda prompt，输入

conda --version

查看Anaconda版本。 如下即为安装完成。

二、创建虚拟环境

        在Anaconda prompt输入

conda create -n ename python=3.X

其中ename即你需要命名的虚拟环境的名称，Python=3.X为虚拟环境中Python的版本。

三、激活虚拟环境

        在Anaconda prompt输入

conda activate ename

ename同样是刚刚命名的环境名称。

四、配置环境

        首先安装numpy，依旧在Anaconda prompt中输入

conda install numpy  

        安装Scipy库

conda install -c anaconda scipy

        查询自己的CUDA版本（目前电脑安装的CUDA版本）

nvcc --version

        查询电脑可以安装的最高版本的CUDA

nvidia-smi

        在Anaconda navigator中激活CUDA

        之后前往pytorch官网，命令行安装对应版本的pytorch，如果命令行不能成功安装，使用下面的代码安装

conda install pytorch torchvision torchaudio pytorch-cuda=X.X -c pytorch -c nvidia

        X.X改成CUDA的版本号，我是11.8。安装完成后，运行程序验证。

import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())

        显示如下即为成功。

        安装opencv库(清华源)

pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple