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
相关推荐
Jamie2019010610 分钟前
健康孪生智能体使用起来复杂吗?医者AI技术核心与用户体验
人工智能
try2find14 分钟前
安装llama-cpp-python踩坑记
开发语言·python·llama
GLAB-Mary15 分钟前
AI会取代网络工程师吗?理解AI在网络安全中的角色
网络·人工智能·web安全
道可云22 分钟前
道可云人工智能每日资讯|浦东启动人工智能创新应用竞赛
人工智能·百度·ar·xr·deepseek
kyle~27 分钟前
目标检测在国防和政府的应用实例
人工智能·目标检测·计算机视觉
兮℡檬,38 分钟前
torchvision中的数据使用
人工智能
Qdgr_1 小时前
价值实证:数字化转型标杆案例深度解析
大数据·数据库·人工智能
c++服务器开发1 小时前
一文详解Character AI:实用指南+ ChatGPT、Gemini对比分析
人工智能·chatgpt
hanniuniu131 小时前
AI时代API挑战加剧,API安全厂商F5护航企业数字未来
人工智能·安全
博观而约取1 小时前
Django ORM 1. 创建模型(Model)
数据库·python·django