cuda python torch 虚拟环境配置

以下是Pytorch和CUDA对应的版本

以下是Pytorch和Python对应的版本

检查cuda与Python版本是否匹配

python 复制代码
import torch
print(torch.__version__) 
print(torch.cuda.is_available())
print(torch.empty(3,4,device='cuda'))

cuda

删除cuda

python 复制代码
conda uninstall cudatoolkit --force
 
conda uninstall cudnn  --force

检查cuda版本

在xshell处检查,cuda版本是10.1

安装10.1版本的cuda:

https://developer.nvidia.com/cuda-toolkit-archive

本文以cuda10.1为例:

在conda虚拟环境中安装CUDA:

python 复制代码
conda install cudatoolkit=10.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

在conda虚拟环境中安装cudnn:

python 复制代码
conda install cudnn=7.6.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

查看cuda版本

在vscode里查看,发现安装的只有10.0

python 复制代码
nvcc -V

由于cuda版本为10.0,Python版本最高只能装到3.6

无论怎么删除Python版本,在(base)(pytorch)等虚拟环境下Python的版本还是3,9,所以干脆新建立了一个虚拟环境

创建新的虚拟环境

python 复制代码
conda create --name env python=3.6

查看Python版本

python 复制代码
python --version

安装torch

查看版本匹配的torch:

Start Locally | PyTorch

python 复制代码
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
相关推荐
FIT2CLOUD飞致云13 分钟前
九月月报丨MaxKB在不同规模医疗机构的应用进展汇报
人工智能·开源
阿里云大数据AI技术15 分钟前
【新模型速递】PAI-Model Gallery云上一键部署Qwen3-Next系列模型
人工智能
袁庭新34 分钟前
全球首位AI机器人部长,背负反腐重任
人工智能·aigc
机器之心1 小时前
谁说Scaling Law到头了?新研究:每一步的微小提升会带来指数级增长
人工智能·openai
该用户已不存在1 小时前
Mojo vs Python vs Rust: 2025年搞AI,该学哪个?
后端·python·rust
算家计算1 小时前
AI配音革命!B站最新开源IndexTTS2本地部署教程:精准对口型,情感随心换
人工智能·开源·aigc
量子位1 小时前
马斯克周末血裁xAI 500人
人工智能·ai编程
算家计算1 小时前
OpenAI最强编程模型GPT-5-Codex发布!可独立编程7小时,编程效率提升10倍
人工智能·ai编程·资讯
站大爷IP3 小时前
Java调用Python的5种实用方案:从简单到进阶的全场景解析
python
聚客AI3 小时前
🌟大模型为什么产生幻觉?预训练到推理的漏洞全揭秘
人工智能·llm·掘金·日新计划