Anaconda安装与conda配置GPU虚拟环境

一、查看和所安装python对应的Anaconda版本并下载

Anaconda 发行说明 — Anaconda 文档

二、进入安装，关键步骤记录，其它默认即可（均为默认）

三、创建虚拟环境

1、查看是否可以使用GPU 及其对应的cuda版本：命令行输入nvidia-smi，有的话会出现一个含有cuda版本的表格

2、安装cuda、cudnn文件（本人安装的cuda11.1）

 参考视频：【yolov5】三部曲系列教程之GPU环境部署_哔哩哔哩_bilibili

 参考文章：在windows11上安装cuda，cudnn，以及GPU版的pytorch_windows11系统pytorch、cuda11.0、cudnn安装-CSDN博客

Cuda和cuDNN安装教程(超级详细)-CSDN博客

Windows10检查Cuda和cuDNN是否安装成功？_查看大cuda时,cmd终端输入nvcc -v命令显示不是内部命令或外部命令,这是为什么-CSDN博客

3、创建conda虚拟环境

conda常用命令：

查看conda版本：conda --version

查看conda的环境配置：conda config --show

创建虚拟环境：conda create -n env_name python=3.8  #python版本为3.8、名字为env_name

查看已安装的虚拟环境：conda info --envs

激活虚拟环境：conda activate env_name

退出虚拟环境：conda deactivate

删除虚拟环境及其中所安装的包：conda remove --name env_name --all

删除虚拟环境中的某个包：conda remove --name env_name  package_name

查看当前环境安装了哪些包：conda list

4、安装GPU版本的pytorch相关库

使用以下命令时会报错，是因为Windows下没有对应的包

pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html

使用该命令安装，更改包的版本

pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html

5、打开pycharm，选择创建的虚拟环境进行测试，输出True

注意：有一个文件在anaconda\library\bin里，也叫conda.bat 用这个就行。

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