cuda10.0配置pytorch1.7.0+monai0.9.0

1. pytorch安装

没有对应cuda10.0的命令，重装10.2嫌麻烦，尝试直接用10.2的cudatoolkit，结果可行。

# CUDA 10.2 （10.0能用，但如果命令里面写cudatoolkit=10.0，pytorch是cpu版的，不好改过来）
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.2 -c pytorch
# 测试
python
import torch
x = torch.rand(5, 3)
print(x)
The output should be something similar to:
tensor([[0.3380, 0.3845, 0.3217],
[0.8337, 0.9050, 0.2650],
[0.2979, 0.7141, 0.9069],
[0.1449, 0.1132, 0.1375],
[0.4675, 0.3947, 0.1426]])
Additionally, to check if your GPU driver and CUDA is enabled and accessible by PyTorch, run the following commands to return whether or not the CUDA driver is enabled:
import torch
torch.cuda.is_available()

2. monai安装
conda install numpy=1.21 (monai依赖numpy>=1.17,pytorch>=1.7.0)
pip install monai
# 测试
python
import monai
monai.config.print_config()

测试通过

cuda10.0配置pytorch1.7.0+monai0.9.0相关推荐

vue-cli3.0配置详解
这次给大家带来vue-cli3.0配置详解,使用vue-cli3.0配置的注意事项有哪些,下面就是实战案例,一起来看一下. 新建项目 1 2 3 4 5 6 7 8 # 安装 npm install ...
揭开.NET 2.0配置之谜（一）
2010-03-20 15:33 by 吴秦, 4828 阅读, 20 评论, 收藏, 编辑此文是译文,原文是Jon Rista,Unraveling the Mysteries of .NET 2 ...
caffe-gpu ubuntu 安装_ubuntu16.04 cuda10.0 配置caffe gpu环境
caffe gpu环境的配置就是一部赤裸裸的血泪史,每次都能遇到很多问题~ 环境依赖 ubuntu 16.04cuda 10.0python 3.5opencv 3.4.10sudo apt-get ...
Ubuntu16.04配置GPU驱动(GTX1060）+ CUDA9.0 + CUDNN7.0.5
1.查看电脑是否有一个可以支持CUDA的GPU a.在终端输入命令行(ctrl+alt+T快捷进入终端): $ lspci | grep -i nvidia 查看电脑GPU信息如下: b.在终端输入命 ...
deepin10.15安装cuda10.1.168 cudnn7.6.1 tensorflow_gpu1.4.0
deepin10.15安装cuda10.1.168 cudnn7.6.1 tensorflow_gpu1.4.0 最近入坑deepin操作系统,基于debian9,和Ubuntu具有一样的操作习惯,由 ...
windows10 1050ti vs2015 openc3.2 cuda8.0配置自己的darknetyolov3
windows10 1050ti vs2015 openc3.2 cuda8.0配置自己的darknetyolov3 本来最近主要学习训练自己maskrcnn训练的事情,昨天突然一时兴起想起来自己的D ...
Ubuntu18.04+cuda10.1+cudnn7.6.3+tensorrt5.1.5.0（deb）
环境配置Ubuntu18.04+cuda10.1+cudnn7.6.3+tensorrt5.1.5.0 Ubuntu18.04安装本系统配置背景 1. cuda10.1安装 2. 驱动安装 3. c ...
mysql 8.0配置主从同步_MySQL8.0.19开启GTID主从同步CentOS8
前言本次搭建目标为1主2从MySQL主从同步结构.采用CentOS8作为操作系统,IP为[10.0.0.211,10.0.0.212,10.0.0.213].MySQL版本为8.0.19,端口均采用3 ...
J2EE 中的服务器 tomcat6.0 配置
Tomcat6.0 配置第一步:下载jdk和tomcat:JDK下载 Tomcat下载最新的jdk为1.6.10,tomcat为6.0,建议jdk1.4以上,tomcat4.0以上第二 ...

cuda10.0配置pytorch1.7.0+monai0.9.0

cuda10.0配置pytorch1.7.0+monai0.9.0相关推荐

最新文章

热门文章