Skip to content

Instantly share code, notes, and snippets.

View wk989898's full-sized avatar
🙃
struggle

nhsmw wk989898

🙃
struggle
View GitHub Profile
package main
import (
"context"
"fmt"
"math/rand"
"time"
"github.com/IBM/sarama"
"github.com/pingcap/log"
@wk989898
wk989898 / conda-pytorch-cuda.md
Last active January 21, 2024 10:54
deepspeed install solution

Install conda cuda

For solving pytorch-version not match system's cuda.

e.g.

ld: cannot find -lcurand: 没有那个文件或目录
collect2: error: ld returned 1 exit status
subprocess.CalledProcessError: Command '['ninja', '-v']' returned non-zero exit status 1.

1. conda install cuda-toolkit and cudnn

@wk989898
wk989898 / useful bash command
Last active March 13, 2024 06:00
useful bash command
# batch kill process
ps aux | grep * | awk '{print $2}' | xargs kill -9
# remove tensorboard files
find ./log -type f -size -1k -exec rm -r {} + && find ./log -empty -exec rm -r {} +