0
点赞
收藏
分享

微信扫一扫

win+rtx3050安装pytorch

慎壹 2022-03-11 阅读 65

在anaconda环境中安装

更换清华源

​
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64
conda config --set show_channel_urls yes
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

​

安装指令(可用)

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1

不需要另外安装cuda和cudnn了,anaconda已经帮我们装好了。

查看python cuda cudnn的版本:

import torch
print(torch.__version__)

import torch
print(torch.version.cuda)
import torch
print(torch.backends.cudnn.version())
举报

相关推荐

0 条评论