AI环境搭建:

AI环境搭建:
0先配置三个环境变量,缺一不可

D:\Anaconda3

D:\Anaconda3\Scripts

D:\Anaconda3\Library\bin

1.查看本机cuda版本

按window按键,并输入NVIDIA control panel------进入界面查看cuda版本;

如果无法正常打开以上步骤,请下载鲁大师,进行显卡驱动安装

2.下载cuda对应的torch和torchvision。软件向下兼容。

下载地址:https://download.pytorch.org/whl/cu102/torch_stable.html

3.安装torch和torchvision.并运行测试代码,测试环境是否搭建正确。

import torch
import torchvision

#查看torch的版本
print(torch.__version__)
#查看torchvision的版本
print(torchvision.__version__)

#查看cuda是否安装成功,成功为True,不成功为False
print(torch.cuda.is_available())
#查看cudnn是否安装成功,成功为True,不成功为False
print(torch.cudnn_is_acceptable(torch.Tensor(1).cuda()))

#查看cuda的版本
print(torch.version.cuda)
#查看cudnn的版本
print(torch.backends.cudnn.version())
#查看当前GPU
print(torch.cuda.get_device_name(0))

4.安装python和对应版本的anaconda步骤省略,记得anconda加入环境变量。

搭建AI环境已经不下5次,好记性不如烂笔头!

你可能感兴趣的:(AI,pytorch,深度学习)