609 字
3 分钟
Python与Pytorch的环境搭建
2024-07-18
统计加载中...

使用VScode搭建Python与Pytorch环境#

1.Anaconda#

  • Anaconda 是一个开源的Python发行版本,其包含了conda、Python等180多个科学包及其依赖项。(详见百科)
  • 总之,当你安装了Anaconda,就不用去Python官网下载解释器了,因为Anaconda内置有Python解释器。
  • 当然了,为确保Anaconda完美安装,请务必将电脑中的Python卸载干净。
  • Anaconda官网下载地址https://www.anaconda.com/download/success

Anaconda下载官网 Anaconda下载选项

  • 安装完成后再配置一下 帐户环境变量

环境变量

2.Cuda (需要Nvidia显卡)#

  • CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。 它包含了CUDA指令集架构(ISA)以及GPU内部的并行计算引擎。(详见百科)
  • 下载地址:https://developer.nvidia.com/cuda-toolkit-archive

Cuda下载界面 Cuda下载选项

  • 之后双击打开,一路点击下一步等待安装完成即可。

3.cuDNN#

  • NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN可以集成到更高级别的机器学习框架中,如谷歌的Tensorflow、加州大学伯克利分校的流行caffe软件。简单的插入式设计可以让开发人员专注于设计和实现神经网络模型,而不是简单调整性能,同时还可以在GPU上实现高性能现代并行计算。
  • 下载地址:https://developer.nvidia.com/cudnn-downloads

cuDNN

4.Pytorch#

  • PyTorch 是一个开源的Python机器学习库,基于Torch库,底层由C++实现,应用于人工智能领域,如计算机视觉和自然语言处理。(详见百科)
  • 下载地址:https://pytorch.org/get-started/locally/

Pytorch下载页面

  • Cuda 根据安装版本来选择,可以向下兼容。
  • 复制给出的下载指令,例如这里给出的是:
  • pip3 install torch torchvision torchaudio —index-url https://download.pytorch.org/whl/cu121
  • 接下来在Windows搜索栏搜索 Anaconda Prompt 打开后将给出的下载指令输入进去。(此时的Pytorch就安装在了整个系统大环境下)

Anaconda Prompt Anaconda Prompt

5.VScode相关设置#

1. 安装python拓展:#

  • 打开VScode拓展,搜索python,安装

Python拓展

2. 设置python环境:#

  • 打开VScode设置,(左下角齿轮->选择设置settings)

齿轮 settings1 settings2

3. 设置python解释器:#

  • 任意打开一个python文件,例如test.py
  • 其中编写代码 print(123)

Python_interpreter_1

python_interpreter_2

python_interpreter_3

  • 至此大功告成
Python与Pytorch的环境搭建
https://blog.srprolin.top/posts/python-1/
作者
RoL1n
发布于
2024-07-18
许可协议
CC BY-NC-SA 4.0