版本:
window10 64位
Anaconda 4.5.11
- 在官网https://pytorch.org/选择要安装的版本,获取安装命令。使用conda命令下载非常慢,所以用了pip命令。安装的pytorch版本为1.0.0
验证:
.成功!
ubuntu+anaconda安装PyTorch-GPU
版本:
Anaconda 4.5.4
nvidia-smi查看显卡信息
nvcc -V查看cuda信息:Cuda compilation tools, release 9.0, V9.0.176
另:
cuda 版本
cat /usr/local/cuda/version.txt
cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
1.新建环境mytorch
conda create -n mytorch python=3.6
source activate mytorch
进入环境目录
2.找到相应版本的安装命令
conda install pytorch torchvision -c pytorch
3.验证