栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

在anaconda环境下正确快速安装pytorch,并在pycharm中实战

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

在anaconda环境下正确快速安装pytorch,并在pycharm中实战

配置:
Windows10系统、已安装Anaconda、具有Invidia显卡(任务管理器查看)

1、点击开始菜单,搜索anaconda prompt

2、进入后输入conda create -n pytorch,创建以pytorch命名的新环境
pytorch可以换成任何你喜欢的名字。

3、conda activate pytorch进入该环境
此时,命令行首的base将变为 pytorch

(重点)4、开始菜单打开cmd,输入nvidia-smi查看自己的cuda版本
这里是CUDA Version: 10.1

(重点)5、进入pytorch官网,ctr+F搜索10.1找到适配版本
10.1换成任意第4步中你的电脑得到的版本
官网链接:https://pytorch.org/get-started/previous-versions/
这是关键,否则装上了也会因为版本适配问题而无法使用。
6、库较大先别着急下载,回到anaconda prompt,添加清华镜像
四行命令:

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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

7、添加之后,拷贝pytorch官网这句话,删去后面的’-c pytorch’
'-c pytorch’的意思是让你从官网下载,国内被墙,下载慢。
只输入这一部分就行:

conda install pytorch== 1.7.0 torchvision== 0.8.0 torchaudio== 0.7.0 cudatoolkit = 10.1

之后开始下载,有y按y。
如果最终报错:Downloaded bytes did not match Content-Length。
可能是你的网络原因也可能是镜像源出现问题,检查并非网络问题后,可以回到第5步,在官网中换一个pytorch版本下载。
8、打开pycharm,创建项目
项目自行创建,不做演示。
接着打开file->settings

如下图,点击右侧齿轮按钮,选Add

左侧选择Conda Environment,勾选Existing environment
第一个Interpreter的框,选择步骤2中,安装pytorch环境的那个位置,里面有python.exe文件
之后Apply、确认即可

9、实战检验
输入代码:

import torch
import torch.nn as nn
from torch.autograd import Variable

print(torch.cuda.is_available())

正确运行,返回true

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/739667.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号