mirrors

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'
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值