Pytorch 安装

nvcc -V 
nvidia-smi

看 CUDA 版本

https://pytorch.org/get-started/previous-versions/ 搜指令。

可以再加下面指令换源:

-i https://pypi.douban.com/simple/
-i https://pypi.tuna.tsinghua.edu.cn/simple/
-i https://mirrors.aliyun.com/pypi/simple/
-i https://pypi.org/simple/

比如:
pytorch==2.1.0+cu118:

pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118
pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118 -i https://pypi.tuna.tsinghua.edu.cn/simple/
posted @ 2023-07-29 20:10  Cold_Chair  阅读(28)  评论(0编辑  收藏  举报