windows10安装pytorch1.3GPU版本(在pycharm及jupyter notebook上运行)

标题windows10安装pytorch1.3GPU版本(在pycharm及jupyter notebook上运行)

前言:由于博主安装多种环境及多个版本的python,导致第一次安装不成功,因此记录本次安装过程,便于读者避坑。

1、安装pytorch准备

首先查看自己电脑的cuda版本,对应pytorch官网的版本要求,如若cuda版本低,可到英伟达官网进行升级。
当确定好安装的版本后,请用清华镜像下载,因为官网下载慢到你想哭(亲身经历)…,清华镜像链接镜像安装

2、建立虚拟环境安装pytorch

建立虚拟环境有利于不同版本的管理,方便调用。
已安装Anaconda3, 打开Anaconda prompt,在第一行输入:

conda create -n pytorch python=3.6

此时创建了pytorch虚拟环境,因为博主的python版本为3.6
等待一会后
会显示需要安装包的信息,会有提示,输入y或n,输入y,进行安装,安装好pytorch虚拟环境后,激活此环境。

conda activate pytorch

结果如下:windows10安装pytorch1.3GPU版本(在pycharm及jupyter notebook上运行)_第1张图片
接下来就是安装pytorch
输入指令

conda install pytorch torchvision cudatoolkit=9.2

此指令很重要,用于清华镜像安装pytorch。由于博主cuda是9…2,不同版本指令不同,参考官网指令。等一段时间,确保网不断,否则重新安装。成功安装好后,及时验证了。
输入指令:
打开cmd,激活python,输入:

import torch
print(torch.__version__) #查看pytorch版本号
print(torch.cuda.is_available()) #查看pytorch安装的cuda版本

如若没有报错,即成功安装。

注意:若报错,请往下看。

已经安装pytorch,依旧没有pytorch模块,可能是环境不对。参考如下:
若直接在cmd中激活python后,输入import torch报错,此时,激活torch环境。

activate pytorch #激活porch虚拟环境
python #打开python
import torch #导入torch模块

结果如图:
windows10安装pytorch1.3GPU版本(在pycharm及jupyter notebook上运行)_第2张图片
可以发现,此时终端可以运行pytorch了!!

3、在jupyter notebook上运行pytorch

如若读者按照上面的步骤在终端可以运行pytorch了,发现在jupyter notebook上依旧显示没有此模块,这时候,是因为自带的jupyter notebook python没有pytorch安装包,此时我们要将之前建立的pytorch虚拟环境移植到jupyter notebook中。
步骤:
打开 Anaconda prompt,在第一行输入:

#pytorch是之前创建的虚拟环境
conda install -n pytorch ipykernel 

等待一会后输入y,安装一些包,安装完后,在jupyter notebook中创建pytorch的kernel名称,指令如下:

#jupyter notebook中就会多了一个pytorch-notebook
python -m ipykernel install --name pytorch-notebook

执行完后,打开jupyter notebook,
windows10安装pytorch1.3GPU版本(在pycharm及jupyter notebook上运行)_第3张图片
此时再import torch,就能成功运行。

4、在pycharm社区版上运行pytorch

步骤:
1、新建项目
windows10安装pytorch1.3GPU版本(在pycharm及jupyter notebook上运行)_第4张图片
2、打开file—>setting–>project Interpreter
按如图所示的建立Interpreter
windows10安装pytorch1.3GPU版本(在pycharm及jupyter notebook上运行)_第5张图片
由于博主已经安装,所以上面红色显示不为空,Location为读者自行定义的虚拟文件夹,其他路径不变。
至此,pycharm可运行pytorch!!!

本文安装参考博客:
https://blog.csdn.net/GXSeveryday/article/details/89976601
https://blog.csdn.net/WJ_MeiMei/article/details/84141521

你可能感兴趣的:(pytorch,pytorch安装,pycharm,jupyter,notebook)