在win10上安装pytorch-gpu版本2

安装anaconda即下载了python,还可以创建虚拟环境。

目录

1.1 anaconda安装

1.2 pytorch-gpu安装


1.1 Anaconda安装

anaconda的安装请看我之前发的tensoflow-gpu安装,里面有详细的安装过程,这里不做重复描述,传送门

1.2 pytorch-gpu安装

anconda下载完成后,打开anaconda prompt,输入

conda create -n pytorch-gpu python=3.7 创建pytorch用的虚拟环境,当然你们虚拟环境也可以不用命名成pytorch-gpu,创建好后,在pytorch官网找到对应的安装包就行下载,我这里直接

复制代码
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

如果安装太慢就需要加上镜像源了

复制代码
永久生效 (pip config set global.index-url+国内镜像地址)
pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/

# 添加Anaconda的TUNA镜像
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 --set show_channel_urls yes

输入这个命令就行下载,如果速度慢可以使用镜像源进行下载,

下载完成之后,在命令行先输入python

再输入

复制代码
import torch
print(torch.__version__)
print(torch.cuda.is_available())

结果成功即为安装成功。

然后在pycharm里面配置anconda的虚拟环境打开file-setting-Python Interpreter-Add添加anaconda的环境

这样就配置好pytorch-gpu版本,并在pycharm上使用。

相关推荐
yangmf204012 分钟前
私有知识库 Coco AI 实战(四):打造 ES 索引参数小助手
大数据·人工智能·elasticsearch·coco ai
IT北辰32 分钟前
Python数据处理:文件的自动化重命名与整合
数据库·python·自动化
大数据在线36 分钟前
当向量数据库与云计算相遇:AI应用全面提速
人工智能·云计算·向量数据库·亚马逊云科技·zilliz
stevenzqzq38 分钟前
编程中如何与AI交互-结构化输入和理解确认机制
人工智能·交互
高峰君主1 小时前
生成式AI全栈入侵:当GPT-4开始自动编写你的Next.js路由时,人类开发者该如何重新定义存在价值?
人工智能
J先生x1 小时前
【开源项目】基于sherpa-onnx的实时语音识别系统 - LiveASR
人工智能·语音识别
沉到海底去吧Go2 小时前
【图片识别改名】批量读取图片区域文字识别后批量改名,基于Python和腾讯云的实现方案
开发语言·python·腾讯云
火星资讯2 小时前
“兴火·燎原”总冠军诞生,云宏信息《金融高算力轻量云平台》登顶
人工智能·科技
百锦再2 小时前
Python深度挖掘:openpyxl和pandas的使用详细
java·开发语言·python·框架·pandas·压力测试·idea
whaosoft-1432 小时前
51c自动驾驶~合集37
人工智能