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

结果如下:激活虚拟环境
接下来就是安装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模块

结果如图:
能够运行torch了
可以发现,此时终端可以运行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,
成功安装
此时再import torch,就能成功运行。

4、在pycharm社区版上运行pytorch

步骤:
1、新建项目
新建项目
2、打开file—>setting–>project Interpreter
按如图所示的建立Interpreter
在这里插入图片描述
由于博主已经安装,所以上面红色显示不为空,Location为读者自行定义的虚拟文件夹,其他路径不变。
至此,pycharm可运行pytorch!!!

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

Jupyter Notebook中使用PyTorch,你需要按照以下步骤进行操作: 1. 首先,在终端中运行命令创建一个虚拟环境,并安装PyTorch和相应的依赖。可以使用以下命令: ``` conda create -n pytorch-gpu python=3.7.0 conda activate pytorch-gpu conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch ``` 2. 如果你觉得通过以上命令安装速度较慢,你可以选择离线下载PyTorch和torchvision的whl文件,然后在虚拟环境中安装。具体步骤如下: - 访问PyTorch官方网站(https://download.pytorch.org/whl/cu113/torch_stable.html)。 - 在页面中搜索你需要的PyTorch和torchvision版本,并下载对应的whl文件。 - 将下载的whl文件传输到你的离线主机(服务器)中。 - 在已经创建好的虚拟环境中,使用pip命令安装whl文件。例如: ``` pip install torch-1.10.0 cu113-cp37-cp37m-linux_x86_64.whl pip install torchvision-0.11.1 cu113-cp37-cp37m-linux_x86_64.whl ``` 3. 在Jupyter Notebook中验证PyTorch是否安装成功。打开Jupyter Notebook,在一个代码单元格中输入以下代码并运行: ``` import torch print(torch.__version__) print(torch.cuda.is_available()) import torchvision print(torchvision.__version__) ``` 这样就完成了在Jupyter Notebook中使用PyTorch安装和验证步骤。你可以在Jupyter Notebook中继续编写和运行PyTorch代码了。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [安装pytorch成功但是在jupyter notebook中无法使用的问题](https://blog.csdn.net/qq_39033580/article/details/124249528)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] - *2* *3* [Jupyter NotebookPycharm代码连接Docker容器中的远程服务器运行](https://blog.csdn.net/qq_43966129/article/details/126843082)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值