CUDA11.0下Pytorch快速安装

小白学一学 2021-07-30

3099

创建虚拟环境方便使用

conda create -n pytorch python=3.x

一、官网安装

1. 进入官网

https://pytorch.org/ 2. 命令行安装

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

二、镜像安装

1. 直接用pip从豆瓣源下载

https://pypi.doubanio.com/simple

2. 记录好要下载的版本

pip install torch==1.7.1 -i https://pypi.doubanio.com/simple --trusted-host pypi.doubanio.com
pip install torchvision==0.8.2 -i https://pypi.douban.com/simple --trusted-host pypi.douban.com
pip install torchaudio==0.7.2 -i https://pypi.douban.com/simple --trusted-host pypi.douban.com

3. 测试pytorch

数据库

文章转载自小白学一学，如果涉嫌侵权，请发送邮件至：contact@modb.pro进行举报，并提供相关证据，一经查实，墨天轮将立刻删除相关内容。

CUDA11.0下Pytorch快速安装

评论