vi ~/.condarc
channels:
- defaults
show_channel_urls: true
default_channels:
- https://mirror.nju.edu.cn/anaconda/pkgs/main
- https://mirror.nju.edu.cn/anaconda/pkgs/r
- https://mirror.nju.edu.cn/anaconda/pkgs/msys2
custom_channels:
conda-forge: https://mirror.nju.edu.cn/anaconda/cloud
pytorch: https://mirror.nju.edu.cn/anaconda/cloud
安装torch
conda install pytorch=1.11 torchvision torchaudio pytorch-cuda=11.3 -c pytorch -c nvidia
2080ti成功安装版本:
>>> import torch
>>> import torchvision
>>> import torchaudio
>>> torch .backends.cudnn.version()
8200
>>> torch.version.cuda
'11.3'
>>> torch.backends.cudnn.is_available()
True
>>> torch.cuda.is_available()
True
>>> torch.__version__
'1.11.0'
>>> torchvision.__version__
'0.12.0'
>>> torchaudio.__version__
'0.11.0'
>>> torch.cuda.get_device_name(0)
'NVIDIA GeForce RTX 2080 Ti'
>>> torch.cuda.get_device_name(1)
'NVIDIA GeForce RTX 2080 Ti'