使用miniconda下载torch库gpu版注意事项

服务器使用miniconda管理环境。
不是很清楚实验室服务器为什么用miniconda来管理环境,麻烦的要死。
使用miniconda来create一个新的环境,该环境只有pip和conda,什么都需要自己配置,注意的是在配置torch等相关的库需要著名服务器上的cuda幸好,不然会conda install cpu版本的库,导致使用不了gpu,例如一下命令

pip3 install torch1.13.0+cu117 torchvision0.14.0+cu117 torchaudio==0.13.0+cu117 -f https://download.pytorch.org/whl/torch_stable.html

你可能感兴趣的:(python,深度学习,pytorch)