AI环境搭建：

会编程的漂亮小姐姐

于 2021-07-27 10:04:17 发布

阅读量599

点赞数

分类专栏： AI 文章标签： pytorch 深度学习

本文链接：https://blog.youkuaiyun.com/u014229742/article/details/119135820

版权

本文详细介绍了AI环境的搭建过程，包括设置Anaconda路径环境变量、检查CUDA版本、根据CUDA版本下载匹配的PyTorch和torchvision、安装验证环境的正确性。强调了记录步骤的重要性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

最低0.47元/天解锁文章