机器学习创建环境

查看 nvida GPU

nvdia-smi

这里主要查看一下cuda version，看一下版本。

安装 pytorch

先创建一个新的环境，然后激活，注意版本不要太新。

mamba install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia
# 这里pytorch-cuda的版本根据cuda version来

安装机器学习的一些环境

mamba install d2l -c conda-forge

d2l 里面封装了许多常用的机器学习库。

不要忘记添加 -c conda-forge 否则可能无法安装成功。