AI env install

2023-10-01

1. 前言

AI的环境就是狗屎

2. cuda toolkit安装

首先进入在自己的电脑上查看驱动版本
可以看到cuda是11.6
https://developer.nvidia.com/cuda-toolkit-archive下载对应cuda的版本
然后安装即可
验证:进入windows的命令行,输入如下命令

nvcc --version

有如下回显即算成功:

3. conda安装

https://www.anaconda.com/下载anaconda
默认安装即可。

4. AI常用库安装

4.1 pytorch安装

由Facebook开发的深度学习框架,提供动态图计算和易于使用的API,广泛用于构建深度神经网络模型
安装时进入anaconda prompt,然后使用如下命令:

conda create -n pytorch_py310 python=3.10
conda activate pytorch_py310
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu116

安装完成后,进入python命令行中,输入如下命令:
出现True说明安装成功。

4.2 opencv安装

开源计算机视觉库,提供了大量用于图像和视频处理的函数和工具,广泛用于计算机视觉任务
进入conda中你想要安装的虚拟环境,然后使用如下命令:

pip install opencv-python

4.3 TenserFlow安装

由Google开发的深度学习框架,提供动态图计算和易于使用的API,广泛用于构建深度神经网络模型
进入conda中你想要安装的虚拟环境,然后使用如下命令:

pip install TensorFlow

5. pycharm安装

so easy
直接官网下社区版即可。

6. jupyter安装

注意,conda中安装jupyter时,需要在你选择的虚拟环境中进行安装,否则会出现jupyter无法启动的情况。
创建虚拟环境安装了pytorch后,再在这个虚拟环境中安装jupyter

conda install jupyter notebook