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
相关推荐
feihui5 分钟前
记一次 Python 服务 TCE 实例进程异常退出排查
python·gunicorn
超龄超能程序猿10 分钟前
使用 Python 对本地图片进行图像分类
开发语言·人工智能·python·机器学习·分类·数据挖掘·scipy
大千AI助手13 分钟前
RLHF:人类反馈强化学习 | 对齐AI与人类价值观的核心引擎
人工智能·深度学习·算法·机器学习·强化学习·rlhf·人类反馈强化学习
谢斯22 分钟前
[python]在drf中使用drf_spectacular
python·django
我爱一条柴ya24 分钟前
【AI大模型】RAG系统组件:向量数据库(ChromaDB)
数据库·人工智能·pytorch·python·ai·ai编程
MARS_AI_29 分钟前
云蝠智能VoiceAgent重构企业电话客服体系
人工智能·自然语言处理·人机交互·交互·信息与通信
天天爱吃肉82181 小时前
从零到一:深度解析汽车标定技术体系与实战策略
python·嵌入式硬件·自动化·汽车
还听珊瑚海吗1 小时前
Python(一)
开发语言·python
LuckyLay2 小时前
1.1.2 运算符与表达式——AI教你学Django
python·django
学不会就看2 小时前
Django--01基本请求与响应流程
后端·python·django