Anaconda + Pytorch +Cuda 环境搭建
Anaconda安装
第一步安装anaconda
Anaconda(官方网站)就是可以便捷获取包且对包能够进行管理,同时对环境可以统一管理的发行版本。Anaconda包含了conda、Python在内的超过180个科学包及其依赖项。
安装教程参考:
https://blog.csdn.net/qq_45281807/article/details/112442577
Cuda安装
CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA™是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。 它包含了CUDA指令集架构(ISA)以及GPU内部的并行计算引擎。
Cuda版本选择:
1.打开NVUDUA控制面板
2.打开系统信息
3.在组件中查看对应的CUDA版本
4.安装Cuda的详细参考:https://blog.csdn.net/XieRuily/article/details/123670141
利用Anaconda安装pytorch
1.创建新环境 我命名为pytorch 可以根据自己的习惯命名
2.通过Navigator打开cmd
如果直接打开cmd则还需要输入
activate 环境名(如:conda activate pytorch)
3.根据自己的安装版本,在Pytorch官网寻找安装命令代码:
(1)Pytorch官网:https://pytorch.org/
如图所示版本则是复制:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
测试
测试conda
在cmd中输入conda -V
测试cuda
在cmd中输入nvcc -V
测试cuda
在cmd中按照以下输入:
- 激活pytorch环境:conda activate pytorch
- 进入python:python
- 测试pytorch:import torch