cuda-pytorch-gpu环境配置

安装准备：windows10、nvidia显卡、cudnn、cuda

一、查看显卡信息版本下载相应的cuda和cudnn,后期改为版本10.0

·直接点击cuda进行安装，一直next下去

·将压缩包解压，放入c盘的cuda根目录下

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0

二、命令操作配置pytorch-gpu环境

  ·pytorch-gpu的环境创建和激活

conda create –n pytorch-gpu python=3.8   ## 创建环境

activate pytorch-gpu  ##激活环境

·pytorch-gpu库的安装

所有的安装都在环境中进行，所以要安装库需要先激活环境  activate pytorch-gpu，然后安装库

# CUDA 11.0
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

下载出错爆红！！！！！可能会因网络出问题

·安装成功之后通过命令python进行验证，返回True预示安装成功

>>> import torch
>>> print(torch.__version__)
1.2.0
>>> print(torch.cuda.is_available())
True
>>> print(torch.version.cuda)
10.0
>>>

 跑深度模型，配置其他依赖库

##其他依赖库
scipy==1.7.1
numpy==1.21.2
matplotlib==3.4.3
opencv_python==4.5.3.56
torch==1.7.1
torchvision==0.8.2
tqdm==4.62.2
Pillow==8.3.2
h5py==2.10.0

为了方便进行安装，将上边的命令复制在requirements.txt文档中，快速安装

pip install -r D:\requirements.txt

·vscode下载配置

查看pytorch版本

>>> import torch
>>> print(torch._version_)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: module 'torch' has no attribute '_version_'
>>> print(torch.__version__)     ##双下划线_ _
1.7.1+cu110
>>>