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
相关推荐
一个java开发14 小时前
distributed.client.Client 用户可调用函数分析
大数据·python
eqwaak014 小时前
Matplotlib 动态显示详解:技术深度与创新思考
网络·python·网络协议·tcp/ip·语言模型·matplotlib
六月的可乐14 小时前
【干货推荐】AI助理前端UI组件-悬浮球组件
前端·人工智能·ui
蔡俊锋14 小时前
【无标题】
人工智能·chatgpt
007php00714 小时前
某大厂MySQL面试之SQL注入触点发现与SQLMap测试
数据库·python·sql·mysql·面试·职场和发展·golang
CodeCraft Studio14 小时前
Excel处理控件Aspose.Cells教程:使用 Python 将 Pandas DataFrame 转换为 Excel
python·json·excel·pandas·csv·aspose·dataframe
说私域14 小时前
基于开源AI大模型AI智能名片S2B2C商城小程序的参与感构建研究
人工智能·小程序·开源
flashlight_hi14 小时前
LeetCode 分类刷题:2563. 统计公平数对的数目
python·算法·leetcode
java1234_小锋14 小时前
Scikit-learn Python机器学习 - 特征预处理 - 归一化 (Normalization):MinMaxScaler
python·机器学习·scikit-learn
码蛊仙尊14 小时前
2025计算机视觉新技术
人工智能·计算机视觉