Linux下配置Pytorch

1.Anaconda

1.1虚拟环境创建

2.Nvidia驱动

3.CUDA驱动安装

4.Pytorch安装

具体的步骤如上:可参考另一位博主的博客非常详细:

Linux服务器配置Python+PyTorch+CUDA深度学习环境_linux cuda环境配置-CSDN博客https://blog.csdn.net/NSJim/article/details/115386936

主要总结一下我在安装中出现的问题:

1.版本冲突问题

pytorch与python 版本不匹配:

初始我创建的conda环境python版本为3.12.3与其适配的cuda版本为11.8.但服务器端通过nviia-smi命令查看结果如下:

最高安装cuda版本为11.7,于是在pytorch找之前版本的pytorch安装。则出现了版本冲突。

解决方法:将python版本换为3.10即可。

相关推荐
小艳加油3 分钟前
Python机器学习与深度学习;Transformer模型/注意力机制/目标检测/语义分割/图神经网络/强化学习/生成式模型/自监督学习/物理信息神经网络等
python·深度学习·机器学习·transformer
学行库小秘2 小时前
ANN神经网络回归预测模型
人工智能·python·深度学习·神经网络·算法·机器学习·回归
Yn3122 小时前
在 Python 中使用 json 模块的完整指南
开发语言·python·json
秋难降2 小时前
线段树的深度解析(最长递增子序列类解题步骤)
数据结构·python·算法
猿榜2 小时前
Python基础-控制结构
python
Ratten2 小时前
【Python 实战】---- 实现一个可选择、配置操作的批量文件上传工具(三)上传类的实现
python
花小璇学linux2 小时前
imx6ull-驱动开发篇24——Linux 中断API函数
linux·驱动开发·嵌入式软件
林开落L3 小时前
库制作与原理(下)
linux·开发语言·centos·库制作与原理
wxy3193 小时前
嵌入式LINUX——————TCP并发服务器
java·linux·网络
阿里云大数据AI技术3 小时前
【跨国数仓迁移最佳实践6】MaxCompute SQL语法及函数功能增强,10万条SQL转写顺利迁移
python·sql