RF-Diffusion 时频扩散无线电信号生成实验复现

文章链接：RF-Diffusion: Radio Signal Generation via Time-Frequency Diffusion
GitHub链接：https://github.com/mobicom24/RF-Diffusion

一、Python环境配置

Windows 11、PyTorch 2.0.1、CUDA 11.8
（1）创建虚拟环境

(base)conda create-n RF-Diffusion python=3.8(base)conda activate RF-Diffusion

（2）查看支持的CUDA版本
Win+R，输入cmd进入控制台。

nvidia-smi

框出来的是当前驱动的CUDA最高支持版本。

（3）下载CUDA 11.8
从NVIDIA官网下载CUDA 11.8安装包，选择对应操作系统版本。
NVIDIA官网链接https://developer.nvidia.com/cuda-11-8-0-download-archive

Installer Type 选择network就行，安装过程中选择精简即可。
验证是否安装成功：

nvcc-V

（4）安装PyTorch 2.0.1
进官网查看之前版本，复制指令安装即可。
官网链接https://pytorch.org/get-started/previous-versions/
指令：

pip install torch==2.0.1torchvision==0.15.2torchaudio==2.0.2--index-url https://download.pytorch.org/whl/cu118

验证是否安装成功：

pythonimporttorch torch.cuda.is_available()exit()