创建Python
环境(需先安装Anaconda3或Miniforge3):
conda create -n joyrl-book python=3.10
conda activate joyrl-book
安装Torch:
# CPU
pip install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1
# CUDA 11.8
pip install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 --index-url https://download.pytorch.org/whl/cu118
# CUDA 12.1
pip install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 --index-url https://download.pytorch.org/whl/cu121
安装其他依赖:
pip install -r requirements.txt
选择感兴趣的*.ipynb
文件,然后使用Jupyter Notebook
打开并运行即可,推荐使用VS Code
的Jupyter
插件