1、下载anaconda
2、创建新环境
conda create -n nas python=3.8.0
3、下载 CUDA 11.7 CUDA Toolkit Archive | NVIDIA Developer
下载 CUDNN https://developer.nvidia.com/cudnn
4、下载 PyTorch(直接下载)
torch==2.0.1
5、下载 https://download.pytorch.org/whl/torch_stable.html (下载到本地之后使用pip install 安装)
torchvision==0.15.2
6、安装 使用pip install cmake==3.28.3 在终端安装
7、安装 aihwkit==0.9.0 (windows比较麻烦,linux系统还简单)
a、在官网下载BLAS OpenBLAS: https://www.openblas.net
b、使用unzip命令解压,cd到BLAS文件夹下面,然后使用命令make激活
在nas环境终端使用pip install aihwkit==0.9.0指令安装。
8、运行程序,需要什么包按照要求下载即可。