机器学习cuda环境配置-离线配置

本人环境：win11 +RTX3060

浅浅记录，怕自己以后安装找不到

先安装anaconda，创建python环境：conda creat  -n yolo python=3.8。

下载离线包：3个

 torch下载地址：https://download.pytorch.org/whl/torch/https://download.pytorch.org/whl/torch/

pytorch下载地址：Index of /anaconda/cloud/pytorch/linux-64/ https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/

cudatoolkit下载：Index of /anaconda/pkgs/main/win-64/ https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/

 注意：torch，pytorch，cudatoolkit的版本必须对应符合，可以参考此博文：版本对应

下载好三个离线包之后，进入存放包的文件夹，输入以下命令即可：

pip install torch-1.10.0+cu113-cp38-cp38-win_amd64.whl
conda install --offline pytorch-1.10.0-py3.8_cuda11.3_cudnn8.2.0_0.tar.bz2
conda install --offline cudatoolkit-11.3.1-h2bc3f7f_2.tar.bz2

三个都安装完毕后，输入以下代码安装可能缺少的包

conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3

最后在输入以下代码查看是否可用

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

可能的报错：

在import torch时，numpy包出现错误，直接pip install numpy

文件夹出错，需要删除文件或者重新下载，解决方法：用管理员身份打开Anaconda Prompt即可。

如果有懒得找版本对应的，可云盘直取：

pan
提取码：g536