wsxk / wsxk.github.io

MIT License
5 stars 0 forks source link

AI env install #157

Open wsxk opened 8 months ago

wsxk commented 8 months ago

https://wsxk.github.io/AI_env/

  1. 前言
  2. cuda toolkit安装
  3. conda安装
  4. pytorch安装
  1. 前言 AI的环境就是狗屎

  2. cuda toolkit安装 首先进入在自己的电脑上查看驱动版本

可以看到cuda是11.6 在https://developer.nvidia.com/cuda-toolkit-archive下载对应cuda的版本 然后安装即可 验证:进入windows的命令行,输入如下命令 nvcc --version

有如下回显即算成功:

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

  2. pytorch安装 进入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说明安装成功。