安装和配置Pytorch和cuda

安装方法

请参考：https://www.cnblogs.com/tryhardwy/p/14659131.html

1. 卸载掉旧版本torch torchvision
2. 先到（https://pytorch.org/get-started/locally/）查到稳定版本torch对应的cuda
3. 下载并安装cuda
4. 到（https://pytorch.org/get-started/locally/）按照对应版本安装torch
5. 安装完成后，进入python验证。

import torch
torch.__version__
torch.cuda.is_available()

显示True则安装成功。

注意事项

注意：直接粘贴

pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/torch_stable.html ，安装的是cpu版！

带cuda版本正确的安装语句是：

pip3 install torch==2.6.0+cu118 torchvision==0.21.0+cu118 --index-url https://download.pytorch.org/whl/cu118

如果上述地址下载太慢，还可以换用国内源例如aliyun（备注：阿里云的torch版本不全，部分最新版本无法下载，经过实验2.1.0版本可以下载并安装）
pip install torch==2.1.0+cu118 --use-deprecated=legacy-resolver --no-cache-dir -f https://mirrors.aliyun.com/pytorch-wheels/cu118

其他的包 如果国内地址下载太慢，可以用以下常用镜像：

常见镜像源

清华：https://pypi.tuna.tsinghua.edu.cn/simple/
阿里云：http://mirrors.aliyun.com/pypi/simple/
中国科技大学：https://pypi.mirrors.ustc.edu.cn/simple/
华中科技大学：http://pypi.hustunique.com/simple/
上海交通大学：https://mirror.sjtu.edu.cn/pypi/web/simple/

如果使用不带https的连接，还需要加上--trusted-host mirrors.xxx.com
例如
pip install numpy<=2.0.0 -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com