pytorch环境配置

本文分享了作者在配置PyTorch环境时遇到的问题,包括使用Anaconda创建虚拟环境、处理版本兼容性(如CUDA与Python版本)、以及安装torch包的注意事项,旨在提供一个完整的配置方法论。
摘要由CSDN通过智能技术生成

在配置pytorch的环境中我遇到了许多问题,踩了不少坑,以下归并一个完整方法论来方便我在下次配置的时候能够顺利些许。

我是用anacoda作为我环境基本存储:

1、关于anacoda,我们需要了解一些他基本的指令:

查看conda版本

conda --version

查看conda的环境配置

conda config --show

创建虚拟环境

conda create -n env_name python=3.8

安装包

pip install 对应的包名

查询已安装包

conda list

在建设虚拟环境中常常遇到版本匹配问题,很重要的是需要知晓显卡算力对应的cuda版本,像30系的显卡就已经不支持11以下的版本了,显卡的信息可以到CUDA 专区 | NVIDIA 开发者上查看包括cuda Toolkit。(当然清华镜像源的方法更较为方便)还有就是Python环境的版本,在近来的代码复现过程中我一直在调整版本,一定要看好对应的Python版本Download Python | Python.org。最后就是在安装torch包,cuda的版本也需要一一对应,选用合适的torch对代码复现更为方便PyTorch

以后遇到的问题再一一写于此处,谢谢。

  • 11
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值