pytorch环境配置

目录

1、Anaconda

安装Anaconda

anaconda常见命令

常用的镜像源

2、确定GPU能装什么版本的CUDA

3、安装CUDA

4、Pytorch安装

CUDA与pytorch版本对应

pytorch安装步骤

5、安装torchvision

6、安装jupyter

7、pycharm的配置

pycharm终端配置,使用Anaconda


1、Anaconda

安装Anaconda

Anaconda下载地址(清华源):Anaconda安装

anaconda常见命令

**·**创建虚拟环境

复制代码
conda create -n 环境名 python=3.7

**·**查看已经创建的虚拟环境

复制代码
conda env list

**·**切换/进入环境

复制代码
activate 环境名

· 删除环境

复制代码
conda remove -n 环境名 -all(退出该环境再进行操作)

**·**复制环境

复制代码
conda create -n 新环境名 --clone 旧环境名

**·**查看包

复制代码
conda list

**·**查看显卡状态及使用情况

复制代码
nvidia-smi

常用的镜像源

阿里云 http://mirrors.aliyun.com/pypi/simple/

中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/

豆瓣 http://pypi.douban.com/simple/

中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/

清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/

2、确定GPU能装什么版本的CUDA

**·**打开任务管理器

打开anaconda的Anaconda Prompt输入命令

复制代码
nvidia-smi

查看GPU能够支持的最高CUDA版本

3、安装CUDA

如果支持CUDA11.x,建议先安装CUDA11.3,下载链接CUDA11.3

输入下列命令查看CUDA是否安装成功

复制代码
nvcc -V

4、Pytorch安装

CUDA与pytorch版本对应

pytorch安装步骤

GPU版本建议离线安装,下载地址:pytorch下载

GPU版本选择cu,后面cp37的意思是python3.7版本,使用离线安装不涉及网络问题。

离线下载的优势是不出错,相对于在线下载方式很有优势。

· 下载好对应的torch后将其放在文件夹内

· 接下来进入存放torch的目标文件夹

· 使用pip命令进行离线下载

输入torch后按Tab键自动补全,后面加上镜像源。提高下载速度

复制代码
pip install torch -i https://pypi.tuna.tsinghua.edu.cn/simple/

· 验证torch是否安装成功

复制代码
//第一步进入python
python
//第二步导包看是否成功
import torch

· 从上述报错可以看出是numpy的问题,利用pip命令更新numpy

复制代码
--退出python
输入Z
--更新numpy
pip install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple/

· 之后再次进入python,查看torch是否下载成功

复制代码
-- 进入python
python
-- 验证torch包
import torch

这样说明torch包下载成功且可以正常导入

· 查看torch版本、查看GPU是否可以用

复制代码
import torch
-- 查看torch版本
print(torch.__version__)
-- 看CPU能不能用,能用返回Ture
print(torch.cuda.is_available())

5、安装torchvision

· torch和torchvision版本对应关系

torchvision下载地址:torchvision下载

找到与CUDA、python、torch相对应的torchvision版本

利用pip离线下载torchvision,输入torchvision后按Tab键即可自动补全路径

复制代码
pip install torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple/

查看torchvision是否安装成功

复制代码
conda list

6、安装jupyter

为了让jupyter使用pytorch,需要在虚拟环境中下载jupyter

复制代码
conda install ipython
conda install jupyter

安装完成后输入

复制代码
jupyter notebook

之后在网页上新建一个python文件,看输出是否正常

7、pycharm的配置

pycharm终端配置,使用Anaconda

可以参考博客:终端配置

更改红色框选的路径为下图cmd.exe后面的内容之后重启pycharm终端即可

之后测试是否配置成功

相关推荐
callJJ1 小时前
Spring Data Redis 两种编程模型详解:同步 vs 响应式
java·spring boot·redis·python·spring
小郑加油1 小时前
python学习Day12:pandas安装与实际运用
开发语言·python·学习
AC赳赳老秦1 小时前
投标合规提效:用 OpenClaw 实现标书 / 合同自动审核、关键词校验、格式优化,降低废标风险
开发语言·前端·python·eclipse·emacs·deepseek·openclaw
.柒宇.1 小时前
AI掘金头条项目-K8s部署实战教程
python·云原生·容器·kubernetes·fastapi
KuaCpp1 小时前
C++面向对象(速过复习版)
开发语言·c++
观北海1 小时前
从 Sim2Sim 到 Sim2Real:以 ONNX 为核心的机器人策略实机落地全指南
python·机器人
wbs_scy1 小时前
Linux线程同步与互斥(三):线程同步深度解析之POSIX 信号量与环形队列生产者消费者模型,从原理到源码彻底吃透
java·开发语言
FL16238631291 小时前
电力设备红外图像与可见光图像配准数据集227对共454张无标注
深度学习
2zcode2 小时前
基于MATLAB元胞自动机(CA)的AZ80A镁合金动态再结晶(DRX)过程模拟
开发语言·matlab·动态再结晶
iCxhust2 小时前
微机原理实践教程(C语言篇)---A001闪烁灯
c语言·开发语言·汇编·单片机·嵌入式硬件·51单片机·微机原理