在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上使用。

相关推荐
CodeCraft Studio43 分钟前
CAD文件处理控件Aspose.CAD教程:使用 Python 将绘图转换为 Photoshop
python·photoshop·cad·aspose·aspose.cad
Python×CATIA工业智造3 小时前
Frida RPC高级应用:动态模拟执行Android so文件实战指南
开发语言·python·pycharm
千宇宙航3 小时前
闲庭信步使用SV搭建图像测试平台:第三十一课——基于神经网络的手写数字识别
图像处理·人工智能·深度学习·神经网络·计算机视觉·fpga开发
onceco3 小时前
领域LLM九讲——第5讲 为什么选择OpenManus而不是QwenAgent(附LLM免费api邀请码)
人工智能·python·深度学习·语言模型·自然语言处理·自动化
狐凄4 小时前
Python实例题:基于 Python 的简单聊天机器人
开发语言·python
悦悦子a啊5 小时前
Python之--基本知识
开发语言·前端·python
jndingxin6 小时前
OpenCV CUDA模块设备层-----高效地计算两个 uint 类型值的带权重平均值
人工智能·opencv·计算机视觉
Sweet锦6 小时前
零基础保姆级本地化部署文心大模型4.5开源系列
人工智能·语言模型·文心一言
笑稀了的野生俊7 小时前
在服务器中下载 HuggingFace 模型:终极指南
linux·服务器·python·bash·gpu算力
Naiva7 小时前
【小技巧】Python+PyCharm IDE 配置解释器出错,环境配置不完整或不兼容。(小智AI、MCP、聚合数据、实时新闻查询、NBA赛事查询)
ide·python·pycharm