我们通过Pytorch官网安装torch的时候,会发现常常由于网速问题安装不成功,下面提供一种简单的方法可以成功安装Cuda12.1,亲测有效。
目录
一、常规方法
二、有效方法
2.1 创建并激活虚拟环境
2.2 添加清华源
2.3 安装torch
一、常规方法
通过Pytorch官网安装torch,网址链接如下:PyTorch
会发现安装速度非常慢,而且很容易失败。
二、有效方法
2.1 创建并激活虚拟环境
创建虚拟环境
conda create -n pytorch39 python==3.9
激活虚拟环境
conda activate pytorch39
2.2 添加清华源
添加清华源可以实现网络加速,大幅度提升安装的成功率
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
2.3 安装torch
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html
验证是否安装成功
Anaconda prompt 里输入 python
>>> import torch
>>> torch.__version__
'2.0.0+cu118'
>>> torch.cuda.is_available()
True
>>> torch.cuda.get_device_name(0)
'NVIDIA GeForce RTX 3090'
如上图所示,pytorch(Cuda12.1)安装成功!