安装torch(GPU版本)并在Pycharm中配置

零.前置环境

1.NVIDIA GPU Computing Toolkit已安装

版本为:11.6

已添加到环境变量

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\libnvvp

在cmd中查看cuda版本

方法1:nvidia-smi

方法2:nvcc -V

2.安装pycharm

我安装的是专业版,自行安装

3.相关信息

windows版本

一、安装Anaconda

anaconda镜像网站

我下载的是:Anaconda3-2023.09-0-Windows-x86_64

一般情况一直next按默认设置,但是注意下面:

(1)选ALL Users

(2)放在 E:\Anaconda\

(3)下面两个可以选,也可以不选,都没大问题,因为我卸载过anaconda所以我全勾上了

安装完成后,在开始界面能看到

二、修改创建新env的地址 + 镜像

(1)env地址设置

因为我想要默认创建的新环境在 E:\Anaconda\envs 下

在下面地址中,找到.condarc文件

C:\Users\15850\.condarc

添加以下内容:

envs_dirs:- E://Anaconda//envs

(2)镜像设置(也可以在anaconda prompt中以命令行设置)

show_channel_urls: true
channels:- http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/- http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/- defaults

 或者在anaconda prompt中以命令行设置

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

三、安装torch

0.版本说明

PS:这步骤可以先不看,先1,等装torch时再来看

因为我的cuda版本是11.6,所以需要查看安装的 pytorch版本是多少在官网查看 

 pytorch官网

因为的cuda版本是11.6并没有显示,所以去 更多版本里查看

更多版本链接

找到我的cuda对应的版本

conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge

pytorch1.12.0

torchvision0.13.0

torchaudio0.12.0

cudatoolkit11.6

python3.8(我想要安装的python版本是3.8)

1.创建环境

打开anacaonda prompt,创建环境并激活,创建了一个名字为:py38,python环境为3.8的conda环境

conda create -n py38 python=3.8

 查看拥有的conda环境,base是默认环境,py38是刚刚创建的环境

conda env list

在文件夹下也能查看文件创建成功

进入到py38这个环境

conda activate py38

2.安装torch(GPU版本) 

根据【0.版本说明】,我们知道了想要下载的版本

torchvision0.13.0

pytorch1.12.0

torchaudio0.12.0

cudatoolkit11.6

那么就在下面的链接下可以下载

pytorch下载镜像

我的cuda版本:11.6,所以在界面中搜索:cu116(快捷键:CTRL+F)

有两种安装方法,一是下载后安装,二是直接在anaconda prompt中安装,下面分别举例介绍

(1)下载后安装(torchvision0.13.0)

下载下来之后,进入到anaconda prompt中安装

文件放在:D:\Code\atools

D:
cd Code
cd atools
pip install "torchvision-0.13.0+cu116-cp38-cp38-win_amd64.whl"

查看是否安装成功

conda list

(2)直接在anaconda prompt中安装 (pytorch1.12.0)

pip install torch==1.12.0+cu116 -f https://download.pytorch.org/whl/torch_stable.html

查看是否安装成功

 安装:torchaudio==0.12.0

pip install torchaudio==0.12.0+cu116 -f https://download.pytorch.org/whl/torch_stable.html

3.在anaconda prompt中测试安装是否成功

python
import torch
torch.__version__
print(torch.cuda.is_available())
exit()

四、在pycharm中配置

可以新建,也可以修改原来代码的编译环境。我在此演示后者

因为conda的环境在下面的文件中

E:\Anaconda\Scripts\conda.exe

所以进行以下设置

 

设置完成

 编写以下代码

import torch
print(torch.cuda.is_available()) # 查看GPU是否可用
print(torch.__version__)

 

运行之后的结果 

PS:新建项目,完成下面的步骤就和上面步骤一样,我就不演示了

 

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.hqwc.cn/news/312740.html

如若内容造成侵权/违法违规/事实不符,请联系编程知识网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

.net8 AOT编绎-跨平台调用C#类库的新方法-函数导出

VB.NET AOT无法编绎DLL,微软的无能,正是你的机会 .net8 AOT编绎-跨平台调用C#类库的新方法-函数导出 1,C#命令行创建工程:dotnet new classlib -o CSharpDllExport 2,编写一个静态方法,并且为它打上UnmanagedCallersO…

Win10取消开机密码

1、首先确认你的 Windows 系统帐户中添加了密码。您可以在设置 - 帐户 - 登录选项中查看电脑是否有添加本机密码。 2、按 WinR 打开运行窗口,在窗口中输入 netplwiz 并按确定。 3、在弹出的用户帐户控制中,我们可以看到当前已添加密码的帐户。把要使用本…

【基础】【Python网络爬虫】【11.字体反扒】认识字体、字体加密(附大量案例代码)(建议收藏)

Python网络爬虫基础 字体反扒1. 认识字体字体概述 2. 处理字体3. 练习案例 - 某小说网址字体解密3.1 请求数据 - 发现问题3.2 下载字体文件3.3 解析字体文件3.4 替换字体 4. 字符串形式字体文件4.1 请求数据 - 发现乱码4.2 下载字体4.3 解析字体4.4 替换数据 5. 字体加密5.1 请…

【Java EE初阶三 】线程的状态与安全(下)

3. 线程安全 线程安全:某个代码,不管它是单个线程执行,还是多个线程执行,都不会产生bug,这个情况就成为“线程安全”。 线程不安全:某个代码,它单个线程执行,不会产生bug&#xff0c…

云计算:OpenStack 分布式架构部署(单控制节点与多计算节点)

目录 一、实验 1.环境 2. 计算服务安装(计算节点2) 3. 网络服务安装(计算节点2) 一、实验 1.环境 (1) 主机 表1 主机 主机架构IP备注controller控制节点192.168.204.210已部署compute01计算节点1192.168.204.211 已部署compute02计算节点2192.168.204.212 (…

EI级 | Matlab实现TCN-BiGRU-Multihead-Attention多头注意力机制多变量时间序列预测

EI级 | Matlab实现TCN-BiGRU-Multihead-Attention多头注意力机制多变量时间序列预测 目录 EI级 | Matlab实现TCN-BiGRU-Multihead-Attention多头注意力机制多变量时间序列预测预测效果基本介绍模型描述程序设计参考资料 预测效果 基本介绍 1.【EI级】 Matlab实现TCN-BiGRU-Mult…

全志R128 DSP开发工具安装教程

资料准备 要编译和仿真DSP,需要以下资料: DSP 核 SDK,SDK 需要包含DSP 编译源码。Cadence Xtensa 的 Windows IDE 工具 (Xplorer‑8.0.13 版本), Windows 版本 DSP 的 package 包。Cadence Xtensa 的 License,用于服…

数据结构:第7章:查找(复习)

目录 顺序查找: 折半查找: 二叉排序树: 4. (程序题) 平衡二叉树: 顺序查找: ASL 折半查找: 这里 j 表示 二叉查找树的第 j 层 二叉排序树: 二叉排序树(Binary Search Tree&…

FDM3D打印系列——RX-78-2高达胸像打印

https://v.youku.com/v_show/id_XNjI4OTQ2NjkyNA.html   大家好,我是阿赵。   2024年的第一篇博客,做一个3D打印作品,RX-78-2高达胸像打印。 成年男人是很少收得到礼物的,所以礼物都要自己准备。这个模型,就算是我…

【时钟】分布式时钟HLC|Logical Time|Vector Clock|True Time

目录 简略 详细 附录 1 分布式系统不能使用NTP的原因 简略 分布式系统中不同于单机系统不能使用NTP(网络时间协议(Network Time Protocol))来获取时间,所以我们需要一个特别的方式来获取分布式系统中的时间,mvcc也是使用time保证读…

数据结构 day6 栈+队列+二分查找+插入排序

插入排序 #include <stdio.h> #include<string.h> #include<stdlib.h> int main(int argc, const char *argv[]) {int a[]{41,50,66,38,32,49,18};int nsizeof(a)/sizeof(a[0]);int i,j,t;for(i1;i<n;i){int ta[i];for(ji-1;j>0;j--){if(t<a[j]){a…

论文阅读: AAAI 2022行人重识别方向论文-PFD_Net

本篇博客用于记录一篇行人重识别方向的论文所提出的优化方法《Pose-Guided Feature Disentangling for Occluded Person Re-identification Based on Transformer》&#xff0c;论文中提出的PDF_Net模型的backbone是采用《TransReID: Transformer-based Object Re-Identificati…