什么是CUDA?
Pytorch安装(GPU版本)
CUDA与cudnn的关系
联想拯救者R9000P 安装torch_GPU 只需要首先安装anconda,创建环境,在该环境下,直接使用一下命令即可:
conda install pytorch torchvision torchaudio cudatoolkit=11.3
或者:
pip install torch==2.0.1+cu117 torchvision==0.15.2+cu117 torchaudio==2.0.2 -f https://download.pytorch.org/whl/torch_stable.html
换镜像源
### 设置清华源镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
## 设置搜索时显示通道地址
conda config --set show_channel_urls yes
### 设置pytorch镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
恢复默认镜像源
conda config --remove-key channels
使用源安装一些包的命令
pip install -i 镜像源 设置信任镜像源 包名
pip install -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com d2l
验证
import torch
torch.cuda.is_available()
True
torch.cuda.current_device()
0
torch.cuda.device(0)
<torch.cuda.device object at 0x0000023034332438>
torch.cuda.device_count()
1
torch.cuda.get_device_name(0)
'GeForce GTX 1660'
查看版本
import torch
torch.__version__
import torchvision
torchvision.__version__